Stars
Training library for Megatron-based models
Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
Scalable data pre processing and curation toolkit for LLMs
Scalable toolkit for efficient model reinforcement
CUDA Templates and Python DSLs for High-Performance Linear Algebra
A machine learning compiler for GPUs, CPUs, and ML accelerators