MatX-inc / seqaxLinks
seqax = sequence modeling + JAX
☆155Updated last month
Alternatives and similar repositories for seqax
Users that are interested in seqax are comparing it to the libraries listed below
Sorting:
- ☆262Updated 10 months ago
- A simple library for scaling up JAX programs☆136Updated 6 months ago
- ☆112Updated last week
- A MAD laboratory to improve AI architecture designs 🧪☆116Updated 5 months ago
- supporting pytorch FSDP for optimizers☆79Updated 5 months ago
- Understand and test language model architectures on synthetic tasks.☆195Updated 2 months ago
- LoRA for arbitrary JAX models and functions☆135Updated last year
- Experiment of using Tangent to autodiff triton☆78Updated last year
- 🧱 Modula software package☆194Updated 2 months ago
- JAX bindings for Flash Attention v2☆88Updated 10 months ago
- ☆78Updated 10 months ago
- Minimal but scalable implementation of large language models in JAX☆34Updated 6 months ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆57Updated 2 years ago
- Named Tensors for Legible Deep Learning in JAX☆176Updated 3 weeks ago
- A library for unit scaling in PyTorch☆125Updated 6 months ago
- JAX implementation of the Llama 2 model☆217Updated last year
- JAX Synergistic Memory Inspector☆172Updated 10 months ago
- A set of Python scripts that makes your experience on TPU better☆53Updated 10 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆392Updated this week
- Cost aware hyperparameter tuning algorithm☆153Updated 11 months ago
- Accelerated First Order Parallel Associative Scan☆180Updated 9 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆133Updated last year
- Train very large language models in Jax.☆204Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆584Updated this week
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆92Updated 5 months ago
- ☆309Updated last week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆126Updated 3 weeks ago
- Inference code for LLaMA models in JAX☆117Updated last year
- nanoGPT-like codebase for LLM training☆94Updated 2 weeks ago
- ☆53Updated last year