lindermanlab / elkLinks
Scalable and Stable Parallelization of Nonlinear RNNS
☆16Updated 4 months ago
Alternatives and similar repositories for elk
Users that are interested in elk are comparing it to the libraries listed below
Sorting:
- ☆32Updated 8 months ago
- ☆53Updated 8 months ago
- Implementation of PSGD optimizer in JAX☆33Updated 5 months ago
- The Energy Transformer block, in JAX☆58Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- ☆104Updated 2 weeks ago
- ☆51Updated last year
- nanoGPT using Equinox☆13Updated 2 years ago
- ☆34Updated 2 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- Flow-matching algorithms in JAX☆97Updated 10 months ago
- Experiments on the impact of depth in transformers and SSMs.☆31Updated 7 months ago
- Turn jitted jax functions back into python source code☆22Updated 6 months ago
- ☆53Updated last year
- ☆29Updated 2 months ago
- ☆31Updated 7 months ago
- ☆9Updated 2 years ago
- 🧱 Modula software package☆200Updated 2 months ago
- Lightning-like training API for JAX with Flax☆41Updated 6 months ago
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆87Updated 3 months ago
- supporting pytorch FSDP for optimizers☆82Updated 6 months ago
- Comparison between GFlowNets & Maximum Entropy RL☆18Updated last year
- Simple Scalable Discrete Diffusion for text in PyTorch☆33Updated 8 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆53Updated 7 months ago
- ☆32Updated last year
- ☆53Updated last year
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- ☆65Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago