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
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• neel-dey / equivariant-gans
  [ICLR 2021] Group Equivariant Generative Adversarial Networks.
  ☆13Updated 3 years ago
• maxxxzdn / implicit-steerable-kernels
  Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
  ☆28Updated 11 months ago
• gcucurull / jax-gat
  JAX implementation of Graph Attention Networks
  ☆13Updated 3 years ago
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆44Updated last year
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆36Updated 2 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• cscarv / riemannian-metric-learning-ot
  ☆17Updated last year
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆24Updated 3 years ago
• mfinzi / residual-pathway-priors
  ☆18Updated 3 years ago
• gcucurull / jax-gcn
  Graph Convolutional Networks in JAX
  ☆31Updated 4 years ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 6 months ago
• gisilvs / AEF
  ☆33Updated last year
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆9Updated 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…
  ☆23Updated last year
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆49Updated last year
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆30Updated 6 months ago
• take-koshizuka / NLSB
  ☆19Updated 2 weeks ago
• francois-rozet / papers-101
  Implementation of papers in 101 lines of code.
  ☆18Updated last year
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆45Updated 2 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆43Updated last year
• necludov / wl-mechanics
  ☆32Updated 8 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆54Updated 11 months ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• cambridge-mlg / neural_diffusion_processes
  ☆13Updated last year
• lucidrains / tranception-pytorch
  Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction
  ☆31Updated 2 years ago
• DiffEqML / kairos
  Model hub for all your DiffeqML needs. Pretrained weights, modules, and basic inference infrastructure
  ☆24Updated last year
• necludov / jam
  Implementation of Action Matching
  ☆38Updated last year