google-deepmind / nanodoLinks
☆289Updated last year
Alternatives and similar repositories for nanodo
Users that are interested in nanodo are comparing it to the libraries listed below
Sorting:
- 🧱 Modula software package☆322Updated 5 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆693Updated last week
- seqax = sequence modeling + JAX☆170Updated 6 months ago
- Minimal yet performant LLM examples in pure JAX☆236Updated 3 weeks ago
- A simple library for scaling up JAX programs☆145Updated 3 months ago
- JAX Synergistic Memory Inspector☆184Updated last year
- Efficient optimizers☆281Updated last month
- Named Tensors for Legible Deep Learning in JAX☆218Updated 2 months ago
- LoRA for arbitrary JAX models and functions☆144Updated last year
- Cost aware hyperparameter tuning algorithm☆179Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆306Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆137Updated last year
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆24Updated last year
- ☆234Updated 11 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆186Updated 2 weeks ago
- Minimal but scalable implementation of large language models in JAX☆35Updated 2 months ago
- Accelerated First Order Parallel Associative Scan☆196Updated 3 weeks ago
- ☆492Updated last year
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆352Updated 2 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆406Updated this week
- JAX implementation of the Llama 2 model☆216Updated 2 years ago
- For optimization algorithm research and development.☆558Updated 3 weeks ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆63Updated 2 years ago
- supporting pytorch FSDP for optimizers☆84Updated last year
- ☆558Updated last year
- Puzzles for exploring transformers☆384Updated 2 years ago
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆334Updated 3 weeks ago
- Understand and test language model architectures on synthetic tasks.☆252Updated 3 weeks ago
- ☆92Updated last year
- A set of Python scripts that makes your experience on TPU better☆56Updated 4 months ago