A library for reproducible deep learning.
MIT License
Statistics for this project are still being loaded, please check back later.
Antares: an automatic engine for multi-platform kernel generation and optimization. Supporting CP...
MSCCL++: A GPU-driven communication stack for scalable AI applications
Building modular LMs with parameter-efficient fine-tuning.
Common PyTorch Modules
Utilities used by the Deep Program Understanding team
System for AI Education Resource.
Tutel MoE: An Optimized Mixture-of-Experts Implementation
A unified wrapper for various ML frameworks - to have one uniform scikit-learn format for predict...
Repo for WWW 2022 paper: Progressively Optimized Bi-Granular Document Representation for Scalable...
Microsoft Linear Algebra Subroutines
FS-Mol is A Few-Shot Learning Dataset of Molecules, containing molecular compounds with measurem...
Generation of protein sequences and evolutionary alignments via discrete diffusion models