google-deepmind / nanodoLinks
β262Updated 10 months ago
Alternatives and similar repositories for nanodo
Users that are interested in nanodo are comparing it to the libraries listed below
Sorting:
- seqax = sequence modeling + JAXβ155Updated last month
- π§± Modula software packageβ194Updated 2 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jaxβ584Updated this week
- β112Updated last week
- Efficient optimizersβ206Updated this week
- A simple library for scaling up JAX programsβ136Updated 6 months ago
- Named Tensors for Legible Deep Learning in JAXβ176Updated 3 weeks ago
- JAX Synergistic Memory Inspectorβ172Updated 10 months ago
- LoRA for arbitrary JAX models and functionsβ135Updated last year
- supporting pytorch FSDP for optimizersβ79Updated 5 months ago
- A MAD laboratory to improve AI architecture designs π§ͺβ116Updated 5 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β126Updated 3 weeks ago
- jax-triton contains integrations between JAX and OpenAI Tritonβ392Updated this week
- Understand and test language model architectures on synthetic tasks.β195Updated 2 months ago
- Cost aware hyperparameter tuning algorithmβ153Updated 11 months ago
- β431Updated 7 months ago
- Implementation of Diffusion Transformer (DiT) in JAXβ276Updated 11 months ago
- β182Updated 5 months ago
- Puzzles for exploring transformersβ347Updated 2 years ago
- β178Updated this week
- For optimization algorithm research and development.β518Updated this week
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 secondsβ237Updated 2 months ago
- Run PyTorch in JAX. π€β245Updated 3 months ago
- β78Updated 10 months ago
- β228Updated 3 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUsβ370Updated last month
- Library for reading and processing ML training data.β444Updated this week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvementβ¦β381Updated this week
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.β287Updated 9 months ago
- β159Updated this week