Alternatives and similar repositories for perceiver-jax:

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

• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 7 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• gcucurull / jax-gat
  JAX implementation of Graph Attention Networks
  ☆13Updated 3 years ago
• lucidrains / tranception-pytorch
  Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction
  ☆31Updated 2 years ago
• cscarv / riemannian-metric-learning-ot
  ☆17Updated last year
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems
  ☆40Updated last week
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆24Updated last year
• take-koshizuka / NLSB
  ☆19Updated last month
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆45Updated 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
• 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 5 months ago
• neel-dey / equivariant-gans
  [ICLR 2021] Group Equivariant Generative Adversarial Networks.
  ☆13Updated 3 years ago
• tychovdo / lila
  Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"
  ☆23Updated 2 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆46Updated last year
• gcucurull / jax-gcn
  Graph Convolutional Networks in JAX
  ☆32Updated 4 years ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆45Updated 2 years ago
• lucidrains / triangle-multiplicative-module
  Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …
  ☆29Updated 3 years ago
• maxxxzdn / implicit-steerable-kernels
  Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
  ☆28Updated 3 weeks ago
• andrew-cr / jump-diffusion
  ☆31Updated 8 months ago
• gregversteeg / esh_dynamics
  Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling
  ☆36Updated 3 years ago
• aidos-lab / dect-evaluation
  Differentiable Euler Characteristic Transform
  ☆17Updated 9 months ago
• lucidrains / equiformer-diffusion
  Implementation of Denoising Diffusion for protein design, but using the new Equiformer (successor to SE3 Transformers) with some addition…
  ☆56Updated 2 years ago
• saleml / continuous-gfn
  ☆24Updated last year
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆20Updated last year
• jw9730 / lps
  [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch
  ☆29Updated 11 months ago
• ngruver / lie-deriv
  ☆37Updated 2 years ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆55Updated last year
• necludov / jam
  Implementation of Action Matching
  ☆39Updated last year