Alternatives and similar repositories for perceiver-jax

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

• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆28Updated 4 months ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆24Updated 2 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆47Updated last year
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆22Updated 4 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆76Updated 3 years ago
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆10Updated 4 years ago
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆22Updated last year
• cscarv / riemannian-metric-learning-ot
  ☆17Updated 2 years ago
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 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.
  ☆40Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 11 months ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆33Updated 8 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆41Updated 6 months ago
• divelab / MFA
  Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.
  ☆17Updated 5 months ago
• facebookresearch / Addressing-the-Topological-Defects-of-Disentanglement
  Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"
  ☆22Updated 2 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆38Updated 2 years ago
• ngruver / lie-deriv
  ☆38Updated 2 years ago
• google-research / gpax
  ☆30Updated 9 months ago
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆24Updated 3 years ago
• facebookresearch / semi-discrete-flow
  code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"
  ☆26Updated 2 years ago
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆30Updated last year
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆46Updated 2 years ago
• gallego-posada / eman
  Equivariant Mesh Attention Networks
  ☆21Updated 2 years ago
• runame / laplace-refinement
  Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks
  ☆10Updated 2 years ago
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆66Updated this week