Highlights
- Pro
Stars
π€ Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
The simplest, fastest repository for training/finetuning medium-sized GPTs.
A toolkit for developing and comparing reinforcement learning algorithms.
The fundamental package for scientific computing with Python.
You like pytorch? You like micrograd? You love tinygrad! β€οΈ
π€ PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
Convert Machine Learning Code Between Frameworks
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
NumPy aware dynamic Python compiler using LLVM
The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.
π A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
A library to generate LaTeX expression from Python code.
Accessible large language models via k-bit quantization for PyTorch.
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv
OpenMMLab's next-generation platform for general 3D object detection.
Google Drive Public File Downloader when Curl/Wget Fails
Fast as FUCK nvim completion. SQLite, concurrent scheduler, hundreds of hours of optimization.
PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
π Accelerate training and inference of π€ Transformers and π€ Diffusers with easy to use hardware optimization tools
AI code-writing assistant that understands data content
OpenMMLab Rotated Object Detection Toolbox and Benchmark