tysam-code / hlb-CIFAR10
Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)
☆1,252Updated 4 months ago
Alternatives and similar repositories for hlb-CIFAR10:
Users that are interested in hlb-CIFAR10 are comparing it to the libraries listed below
- Cramming the training of a (BERT-type) language model into limited compute.☆1,331Updated 10 months ago
- Tensors, for human consumption☆1,247Updated 5 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,769Updated last week
- Fast & Simple repository for pre-training and fine-tuning T5-style models☆1,003Updated 8 months ago
- What would you do with 1000 H100s...☆1,043Updated last year
- Convolutions for Sequence Modeling☆878Updated 10 months ago
- Puzzles for exploring transformers☆344Updated 2 years ago
- 🤖 A PyTorch library of curated Transformer models and their composable components☆885Updated last year
- Notebooks and various random fun☆1,096Updated 2 years ago
- ☆430Updated 6 months ago
- maximal update parametrization (µP)☆1,500Updated 9 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆379Updated 2 weeks ago
- Schedule-Free Optimization in PyTorch☆2,150Updated 3 weeks ago
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆957Updated last year
- TensorDict is a pytorch dedicated tensor container.☆920Updated this week
- Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"☆479Updated 2 years ago
- ☆776Updated last week
- Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.☆381Updated 11 months ago
- arxiv-sanity lite: tag arxiv papers of interest get recommendations of similar papers in a nice UI using SVMs over tfidf feature vectors …☆1,276Updated last year
- Language Modeling with the H3 State Space Model☆520Updated last year
- Everything you want to know about Google Cloud TPU☆527Updated 9 months ago
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆345Updated 9 months ago
- ☆531Updated last year
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,040Updated last year
- For optimization algorithm research and development.☆509Updated this week
- Annotated version of the Mamba paper☆483Updated last year
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,390Updated this week
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,566Updated last year
- A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.☆1,269Updated 2 weeks ago
- Implementation of Hinton's forward-forward (FF) algorithm - an alternative to back-propagation☆1,480Updated last year