Alternatives and similar repositories for reinforce

Users that are interested in reinforce are comparing it to the libraries listed below

• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆90Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆88Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆36Updated last year
• Benjamin-Walker / selective-ssms-and-linear-cdes
  Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)
  ☆15Updated 9 months ago
• radarFudan / mamba-minimal-jax
  ☆33Updated 10 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆38Updated last month
• samblouir / birdie
  ☆13Updated 4 months ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆81Updated last year
• srush / mamba-primer
  ☆38Updated last year
• kelechi-c / dit_flow
  DiT (training + flow matching) in Jax
  ☆10Updated 9 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆60Updated 7 months ago
• google-deepmind / spectral_ssm
  ☆34Updated last year
• ag1988 / dlr
  The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…
  ☆23Updated 2 years ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 8 months ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆35Updated 11 months ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• Gothos / LRU-pytorch
  Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch
  ☆71Updated 5 months ago
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆34Updated 2 years ago
• lucidrains / ppo
  An implementation of PPO in Pytorch
  ☆96Updated 2 months ago
• shikaiqiu / compute-better-spent
  ☆58Updated last year
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆60Updated 2 years ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆54Updated 3 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated last year
• romanpogodin / towards-bio-plausible-conv
  ☆14Updated 4 years ago
• google-deepmind / zipfian_environments
  ☆28Updated 3 years ago
• kvfrans / splus
  ☆120Updated 3 months ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆34Updated 2 years ago
• dtunai / Griffin-Jax
  Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"
  ☆14Updated last year