Alternatives and similar repositories for reinforce:

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

• mohmdelsayed / HesScale
  Scalable Computation of Hessian Diagonals
  ☆13Updated 10 months ago
• enajx / HyperNCA
  ☆38Updated 2 years ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆87Updated 9 months ago
• ag1988 / dlr
  The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…
  ☆20Updated 2 years ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆32Updated 5 months ago
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆30Updated last year
• radarFudan / mamba-minimal-jax
  ☆30Updated 4 months ago
• Gothos / LRU-pytorch
  Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch
  ☆68Updated last year
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆37Updated 3 months ago
• opallab / positional_attention
  Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"
  ☆14Updated 2 months ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 10 months ago
• srush / mamba-primer
  ☆37Updated last year
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆32Updated last year
• okarthikb / state-space-models
  ☆27Updated 9 months ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆57Updated last year
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆31Updated 9 months ago
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆52Updated 5 months ago
• SuReLI / NeurOps
  Implementations of growing and pruning in neural networks
  ☆22Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 2 months ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆33Updated 8 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 8 months ago
• wilson1yan / cs294-158-ssl
  ☆14Updated 2 years ago
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆49Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆55Updated 2 months ago
• srush / Tensor-Puzzles-Penzai
  ☆20Updated 11 months ago
• yash-srivastava19 / arrakis
  Arrakis is a library to conduct, track and visualize mechanistic interpretability experiments.
  ☆26Updated last month