MatX-inc / seqaxLinks
seqax = sequence modeling + JAX
โ159Updated last week
Alternatives and similar repositories for seqax
Users that are interested in seqax are comparing it to the libraries listed below
Sorting:
- โ270Updated 11 months ago
- A MAD laboratory to improve AI architecture designs ๐งชโ120Updated 6 months ago
- โ78Updated 11 months ago
- โ122Updated last month
- A simple library for scaling up JAX programsโ139Updated 7 months ago
- ๐งฑ Modula software packageโ200Updated 2 months ago
- JAX bindings for Flash Attention v2โ89Updated 11 months ago
- A library for unit scaling in PyTorchโ125Updated 6 months ago
- supporting pytorch FSDP for optimizersโ82Updated 6 months ago
- Minimal but scalable implementation of large language models in JAXโ35Updated 7 months ago
- Experiment of using Tangent to autodiff tritonโ79Updated last year
- LoRA for arbitrary JAX models and functionsโ138Updated last year
- Understand and test language model architectures on synthetic tasks.โ217Updated last week
- JAX implementation of the Llama 2 modelโ218Updated last year
- Named Tensors for Legible Deep Learning in JAXโ180Updated last month
- jax-triton contains integrations between JAX and OpenAI Tritonโ400Updated 2 weeks ago
- Inference code for LLaMA models in JAXโ118Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.โ134Updated this week
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"โ235Updated 2 weeks ago
- โ186Updated 2 weeks ago
- A set of Python scripts that makes your experience on TPU betterโ55Updated 11 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jaxโ595Updated this week
- Accelerated First Order Parallel Associative Scanโ182Updated 10 months ago
- JAX Synergistic Memory Inspectorโ174Updated 11 months ago
- Train very large language models in Jax.โ204Updated last year
- Two implementations of ZeRO-1 optimizer sharding in JAXโ14Updated 2 years ago
- โ53Updated last year
- Implementation of PSGD optimizer in JAXโ33Updated 5 months ago
- Jax/Flax rewrite of Karpathy's nanoGPTโ57Updated 2 years ago
- โ60Updated 3 years ago