NVlabs/vibetensor

You like pytorch? You like micrograd? You love tinygrad! ❤️

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A coding-free framework built on PyTorch for reproducible deep learning studies. PyTorch Ecosystem. 🏆26 knowledge distillation methods presented at TPAMI, CVPR, ICLR, ECCV, NeurIPS, ICCV, AAAI, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

Build high-performance AI models with modular building blocks

The fastai deep learning library

Deep-Learning framework for Engineering AI. Built on transformer building blocks, it delivers the full engineering stack, allowing teams to build, train, and operate industrial simulation models across engineering verticals.

nCPU: model-native and tensor-optimized CPU research runtimes with organized workloads, tools, and docs