Alternatives and similar repositories for reinforce

Users that are interested in reinforce 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
  ☆92Updated last year
• srush / mamba-primer
  ☆38Updated last year
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• mohmdelsayed / HesScale
  Scalable Computation of Hessian Diagonals
  ☆14Updated last year
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆91Updated last year
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 10 months ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 3 years ago
• google-deepmind / spectral_ssm
  ☆34Updated last year
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆34Updated 2 years ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated last year
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆32Updated 3 weeks ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆44Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆37Updated 2 years ago
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆60Updated this week
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆113Updated 11 months ago
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• eemlcommunity / PracticalSessions2023
  ☆69Updated 2 years ago
• samblouir / birdie
  ☆13Updated this week
• Benjamin-Walker / selective-ssms-and-linear-cdes
  Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)
  ☆15Updated 11 months ago
• kelechi-c / dit_flow
  DiT (training + flow matching) in Jax
  ☆10Updated 11 months ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆37Updated 2 years ago
• apple / ml-pararnn
  ☆153Updated last month
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆83Updated last year
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated last year
• kvfrans / splus
  ☆122Updated 6 months ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆125Updated last year
• enajx / HyperNCA
  ☆41Updated 3 years ago
• erfanzar / jax-flash-attn2
  A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…
  ☆32Updated 9 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last month