Alternatives and similar repositories for 2023-transformers-rotf

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

• KindXiaoming / BIMT
  Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
  ☆173Updated 2 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆88Updated last year
• luchris429 / DiscoPOP
  Code for Discovering Preference Optimization Algorithms with and for Large Language Models
  ☆63Updated last year
• AndPotap / einsum-search
  ☆33Updated last year
• speed1313 / jaxkan
  JAX implementation of Kolmogorov Arnold Networks (KANs).
  ☆10Updated last year
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆81Updated last year
• AllanYangZhou / universal_neural_functional
  ☆52Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 11 months ago
• shikaiqiu / compute-better-spent
  ☆58Updated last year
• radarFudan / mamba-minimal-jax
  ☆33Updated 10 months ago
• PolymathicAI / xVal
  Repository for code used in the xVal paper
  ☆144Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆143Updated 11 months ago
• borjanG / 2023-transformers
  Codes for the paper The emergence of clusters in self-attention dynamics.
  ☆17Updated last year
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆58Updated last year
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆74Updated 2 years ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆129Updated 9 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆54Updated 3 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆35Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆72Updated 3 months ago
• enajx / HyperNCA
  ☆41Updated 3 years ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆30Updated 2 years ago
• google-deepmind / spectral_ssm
  ☆34Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 10 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• nocotan / awesome-information-geometry
  About A collection of AWESOME things about information geometry Topics
  ☆166Updated last year
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆100Updated last year
• jxbz / agd
  Automatic gradient descent
  ☆213Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• kvfrans / splus
  ☆120Updated 3 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆142Updated last year