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
  ☆23Updated 2 years ago
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆78Updated 3 years ago
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆29Updated 6 months ago
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 years ago
• google-deepmind / annealed_flow_transport
  ☆51Updated 2 years ago
• ghliu / DeepGSB
  Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral
  ☆52Updated 2 years ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆65Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆25Updated 3 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆51Updated 2 years ago
• necludov / wl-mechanics
  ☆31Updated last year
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆56Updated 2 years ago
• cscarv / riemannian-metric-learning-ot
  ☆17Updated 2 years ago
• tychovdo / noethers-razor
  Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…
  ☆10Updated 11 months ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆47Updated 3 years ago
• take-koshizuka / NLSB
  ☆20Updated 7 months ago
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆36Updated 11 months ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆47Updated 2 years ago
• gregversteeg / esh_dynamics
  Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling
  ☆36Updated 3 years ago
• vdutor / neural-diffusion-processes
  Neural Diffusion Processes
  ☆81Updated last year
• ngruver / lie-deriv
  ☆38Updated 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
• zdhNarsil / Diffusion-Generative-Flow-Samplers
  PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…
  ☆25Updated last year
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆31Updated 2 years ago
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆22Updated 4 years ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated last year
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆66Updated 6 months ago