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
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆47Updated last year
• necludov / jam
  Implementation of Action Matching
  ☆41Updated last year
• 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
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆29Updated 2 months ago
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• gregversteeg / esh_dynamics
  Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling
  ☆36Updated 3 years ago
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆76Updated 3 years ago
• 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.
  ☆24Updated 2 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆46Updated last year
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆21Updated last year
• TianrongChen / DMSB
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆25Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months 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
• take-koshizuka / NLSB
  ☆19Updated 3 months ago
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆51Updated this week
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• GFNOrg / gfn-diffusion
  ☆33Updated last month
• 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 7 months ago
• necludov / wl-mechanics
  ☆32Updated 11 months ago
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆30Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 9 months ago
• lollcat / se3-augmented-coupling-flows
  SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.
  ☆24Updated last year
• VincentStimper / resampled-base-flows
  Normalizing Flows with a resampled base distribution
  ☆46Updated 2 years ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆33Updated 7 months ago
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆24Updated 2 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆38Updated 2 years ago