kvfrans / splusLinks
☆119Updated 5 months ago
Alternatives and similar repositories for splus
Users that are interested in splus are comparing it to the libraries listed below
Sorting:
- A simple library for scaling up JAX programs☆144Updated last week
- Implementation of PSGD optimizer in JAX☆35Updated 10 months ago
- Flow-matching algorithms in JAX☆106Updated last year
- LoRA for arbitrary JAX models and functions☆142Updated last year
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Updated last year
- ☆34Updated 11 months ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆115Updated 3 weeks ago
- Pytorch-like dataloaders for JAX.☆96Updated 5 months ago
- A simple, performant and scalable JAX-based world modeling codebase☆100Updated last week
- Minimal but scalable implementation of large language models in JAX☆35Updated 2 months ago
- If it quacks like a tensor...☆59Updated last year
- Run PyTorch in JAX. 🤝☆306Updated last month
- Accelerated First Order Parallel Associative Scan☆189Updated last year
- Supporting code for the blog post on modular manifolds.☆101Updated last month
- ☆53Updated last year
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆39Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆105Updated 3 weeks ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆89Updated last year
- Minimal yet performant LLM examples in pure JAX☆198Updated last month
- Minimal, lightweight JAX implementations of popular models.☆117Updated this week
- ☆68Updated 11 months ago
- supporting pytorch FSDP for optimizers☆83Updated 11 months ago
- ☆221Updated 11 months ago
- 📄Small Batch Size Training for Language Models☆63Updated last month
- WIP☆93Updated last year
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆21Updated last month
- Lightning-like training API for JAX with Flax☆44Updated 11 months ago
- ☆61Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆62Updated 2 years ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆20Updated 2 years ago