Alternatives and similar repositories for 2023-transformers-rotf

Users that are interested in 2023-transformers-rotf are comparing it to the libraries listed below

• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆81Updated last year
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆78Updated last year
• shikaiqiu / compute-better-spent
  ☆53Updated 8 months ago
• radarFudan / mamba-minimal-jax
  ☆29Updated 6 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• google-deepmind / spectral_ssm
  ☆32Updated last year
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆86Updated 2 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆83Updated last week
• AndPotap / einsum-search
  ☆32Updated 8 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆30Updated 2 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆92Updated 9 months ago
• luchris429 / DiscoPOP
  Code for Discovering Preference Optimization Algorithms with and for Large Language Models
  ☆62Updated 11 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 11 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆53Updated last month
• deep-symbolic-mathematics / Multimodal-Math-Pretraining
  [ICLR 2024 Spotlight] This is the official code for the paper "SNIP: Bridging Mathematical Symbolic and Numeric Realms with Unified Pre-t…
  ☆52Updated 7 months ago
• facebookresearch / LAWT
  Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)
  ☆68Updated 9 months ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 5 months ago
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆99Updated 9 months ago
• sail-sg / stde
  Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024
  ☆108Updated 6 months ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆78Updated 2 years ago
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆124Updated last year
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆13Updated 2 years ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆116Updated 5 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 7 months ago
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆72Updated 2 years ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆57Updated last year
• srush / mamba-primer
  ☆37Updated last year
• speed1313 / jaxkan
  JAX implementation of Kolmogorov Arnold Networks (KANs).
  ☆10Updated last year