Alternatives and similar repositories for perceiver-jax

Users that are interested in perceiver-jax are comparing it to the libraries listed below

• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆25Updated 2 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆47Updated 2 years ago
• necludov / jam
  Implementation of Action Matching
  ☆48Updated 2 years ago
• DoronHav / WassersteinFlowMatching
  ☆35Updated last week
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆78Updated 4 years ago
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆23Updated 5 years ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆50Updated 3 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆52Updated 2 years ago
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• cscarv / riemannian-metric-learning-ot
  ☆18Updated 2 years ago
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆29Updated 9 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆66Updated last year
• ghliu / DeepGSB
  Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral
  ☆52Updated 3 years ago
• vdutor / neural-diffusion-processes
  Neural Diffusion Processes
  ☆81Updated last year
• take-koshizuka / NLSB
  ☆21Updated 10 months ago
• saleml / continuous-gfn
  ☆27Updated 2 years ago
• gregversteeg / esh_dynamics
  Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling
  ☆36Updated 4 years ago
• necludov / wl-mechanics
  ☆31Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆44Updated last year
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆57Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆38Updated 3 years ago
• runame / laplace-refinement
  Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks
  ☆10Updated 3 years ago
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 5 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆55Updated 4 years ago
• aidos-lab / dect-evaluation
  Differentiable Euler Characteristic Transform
  ☆17Updated last year
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆31Updated 2 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 4 years ago
• ngruver / lie-deriv
  ☆39Updated 3 years ago