Library for OCR-related tasks powered by Deep Learning
Uplift modeling and causal inference with machine learning algorithms
AIMET is a library that provides advanced quantization and compression
A Pythonic framework to simplify AI service building
A Unified Library for Parameter-Efficient Learning
Python Package for ML-Based Heterogeneous Treatment Effects Estimation
Easy-to-use Speech Toolkit including Self-Supervised Learning model
LLM training code for MosaicML foundation models
OpenFieldAI is an AI based Open Field Test Rodent Tracker
High quality, fast, modular reference implementation of SSD in PyTorch
Toolbox of models, callbacks, and datasets for AI/ML researchers
Implementation of model parallel autoregressive transformers on GPUs
Sequence-to-sequence framework, focused on Neural Machine Translation