NVlabs / Forecasting-Model-Search
A system for automating selection and optimization of pre-trained models from the TAO Model Zoo
☆22Updated 4 months ago
Related projects ⓘ
Alternatives and complementary repositories for Forecasting-Model-Search
- ☆46Updated last month
- Code for minimum-entropy coupling.☆30Updated 4 months ago
- ☆48Updated last week
- Triton Implementation of HyperAttention Algorithm☆46Updated 11 months ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆17Updated 2 weeks ago
- Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead☆113Updated this week
- ☆128Updated this week
- ☆48Updated 9 months ago
- ☆39Updated 10 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆29Updated 2 weeks ago
- ☆53Updated 10 months ago
- An implementation of the Llama architecture, to instruct and delight☆21Updated 3 months ago
- ☆76Updated 7 months ago
- LLM training in simple, raw C/CUDA☆12Updated last month
- ☆53Updated 3 weeks ago
- Optimization algorithm which fits a ResNet to CIFAR-10 5x faster than SGD / Adam (with terrible generalization)☆12Updated last year
- ☆29Updated 2 months ago
- Utilities for PyTorch distributed☆23Updated last year
- Experiment of using Tangent to autodiff triton☆72Updated 9 months ago
- ☆35Updated 7 months ago
- Collection of autoregressive model implementation☆67Updated this week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆84Updated this week
- Scalable neural net training via automatic normalization in the modular norm.☆121Updated 3 months ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆36Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆70Updated 6 months ago
- Jax like function transformation engine but micro, microjax☆26Updated 3 weeks ago
- ☆30Updated 6 months ago
- ☆28Updated 7 months ago
- ☆16Updated last month