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
  ☆77Updated 3 years ago
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 years ago
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆29Updated 6 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆62Updated 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
  ☆47Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 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.
  ☆42Updated last year
• ghliu / DeepGSB
  Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral
  ☆51Updated 2 years ago
• gcucurull / jax-gcn
  Graph Convolutional Networks in JAX
  ☆33Updated 4 years ago
• cscarv / riemannian-metric-learning-ot
  ☆17Updated 2 years ago
• ngruver / lie-deriv
  ☆38Updated 2 years ago
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆56Updated 2 years ago
• mcbal / deep-implicit-attention
  Implementation of deep implicit attention in PyTorch
  ☆65Updated 4 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆47Updated 2 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆50Updated 3 years ago
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆25Updated 3 years ago
• matbun / EBM--Generative-Energy-Based-Modeling
  Energy Based Models are a quite novel technique for density estimation. In this university project I explore this new research topic and …
  ☆16Updated 4 years ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆50Updated 2 years ago
• lollcat / fab-jax
  Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.
  ☆11Updated last year
• noamroze / moser_flow
  ☆17Updated 3 years ago
• divelab / MFA
  Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.
  ☆17Updated 7 months ago
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• take-koshizuka / NLSB
  ☆20Updated 6 months 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 10 months ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆52Updated 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