A series of math-specific large language models of our Qwen2 series
Implementation of the Surya Foundation Model for Heliophysics
A modular high-level library to train embodied AI agents
LLM training code for MosaicML foundation models
The standard data-centric AI package for data quality and ML
Leveraging BERT and c-TF-IDF to create easily interpretable topics
NVIDIA Federated Learning Application Runtime Environment
Easy-to-use and powerful NLP library with Awesome model zoo
Data Lake for Deep Learning. Build, manage, and query datasets
A machine learning library for detecting anomalies in signals
Simplest working implementation of Stylegan2
Bring the notion of Model-as-a-Service to life
An MLOps framework to package, deploy, monitor and manage models
Proofs, cases, concept supplements, and reference explanations
Training data (data labeling, annotation, workflow) for all data types
A refreshing functional take on deep learning
Tool for visualizing and tracking your machine learning experiments
AutoGluon: AutoML for Image, Text, and Tabular Data
A PyTorch-based Speech Toolkit
Create HTML profiling reports from pandas DataFrame objects
Library for serving Transformers models on Amazon SageMaker
Train machine learning models within Docker containers
Best practices on recommendation systems
The AI framework that adds the engineering to prompt engineering
Debug, evaluate, and monitor your LLMapps, RAG systems, and agentic AI