sooheon / perceiver-jax

Related projects ⓘ

Alternatives and complementary repositories for perceiver-jax

• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆43Updated last year
• ngruver / lie-deriv
  ☆36Updated 2 years ago
• gcucurull / jax-gat
  JAX implementation of Graph Attention Networks
  ☆13Updated 2 years ago
• lucidrains / tranception-pytorch
  Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction
  ☆31Updated 2 years ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆48Updated 3 months ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆35Updated last year
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆51Updated 8 months ago
• maxxxzdn / implicit-steerable-kernels
  Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
  ☆26Updated 8 months ago
• neel-dey / equivariant-gans
  [ICLR 2021] Group Equivariant Generative Adversarial Networks.
  ☆13Updated 3 years ago
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Updated 3 years ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆23Updated last year
• gallego-posada / eman
  Equivariant Mesh Attention Networks
  ☆21Updated 2 years ago
• danielegrattarola / GINR
  Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).
  ☆67Updated last year
• gisilvs / AEF
  ☆33Updated last year
• merlresearch / partial_gcnn
  Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"
  ☆10Updated 2 years ago
• tychovdo / lila
  Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"
  ☆23Updated 2 years ago
• TianrongChen / DMSB
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆20Updated last year
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆21Updated 3 years ago
• gengala / egnca
  Official implementation of E(n)-equivariant Graph Neural Cellular Automata
  ☆24Updated 7 months ago
• ghliu / DeepGSB
  Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral
  ☆45Updated 2 years ago
• georgosgeorgos / trajectory-alignment-diffusion
  Code for "Aligning Optimization Trajectories with Diffusion Models for Constrained Design Generation" @ NeurIPS 2023
  ☆9Updated last year
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆42Updated last year
• cscarv / riemannian-metric-learning-ot
  ☆16Updated last year
• jmtomczak / git_flow
  General Invertible Transformations for Flow-based Generative Models
  ☆17Updated 3 years ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆38Updated 9 months ago
• Zymrael / torchSODE
  PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization
  ☆16Updated 4 years ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆28Updated last month
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago