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
  ☆24Updated last year
• necludov / jam
  Implementation of Action Matching
  ☆41Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• take-koshizuka / NLSB
  ☆19Updated 2 months ago
• maxxxzdn / implicit-steerable-kernels
  Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
  ☆28Updated last month
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆37Updated 2 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆47Updated last year
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• cscarv / riemannian-metric-learning-ot
  ☆17Updated last year
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆45Updated last year
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• TianrongChen / DMSB
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆25Updated last year
• gregversteeg / esh_dynamics
  Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling
  ☆36Updated 3 years ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆45Updated 2 years ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆57Updated last year
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆21Updated 4 years ago
• saleml / continuous-gfn
  ☆24Updated last year
• necludov / wl-mechanics
  ☆32Updated 10 months ago
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆21Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 4 months ago
• DiffEqML / kairos
  Model hub for all your DiffeqML needs. Pretrained weights, modules, and basic inference infrastructure
  ☆24Updated 2 years ago
• ngruver / lie-deriv
  ☆37Updated 2 years ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆49Updated last year
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• ghliu / DeepGSB
  Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral
  ☆47Updated 2 years ago
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems
  ☆44Updated last week
• 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
• danielegrattarola / GINR
  Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).
  ☆70Updated 2 years ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆33Updated 8 months ago
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆76Updated 3 years ago