msaroufim / mynotesLinks
☆18Updated last month
Alternatives and similar repositories for mynotes
Users that are interested in mynotes are comparing it to the libraries listed below
Sorting:
- See https://github.com/cuda-mode/triton-index/ instead!☆11Updated last year
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated last year
- ML/DL Math and Method notes☆65Updated 2 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆39Updated 3 weeks ago
- ☆17Updated 2 years ago
- ☆13Updated this week
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 3 months ago
- ☆19Updated 3 years ago
- An implementation of the Llama architecture, to instruct and delight☆21Updated 6 months ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- Functional local implementations of main model parallelism approaches☆95Updated 2 years ago
- Annotations of the interesting ML papers I read☆266Updated 2 months ago
- Code associated to papers on superposition (in ML interpretability)☆33Updated 3 years ago
- Supercharge huggingface transformers with model parallelism.☆77Updated 4 months ago
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆87Updated 2 years ago
- Various transformers for FSDP research☆38Updated 3 years ago
- Rax is a Learning-to-Rank library written in JAX.☆335Updated 3 months ago
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆92Updated 2 years ago
- NLP Examples using the 🤗 libraries☆40Updated 4 years ago
- ☆91Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆297Updated last year
- NLP with Rust for Python 🦀🐍☆70Updated 7 months ago
- Resources from the EleutherAI Math Reading Group☆54Updated 9 months ago
- Experiment of using Tangent to autodiff triton☆81Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆195Updated 6 months ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Neural Networks for JAX☆84Updated last year