A simple but complete full-attention transformer
PyTorch building blocks for the OLMo ecosystem
150+ quantitative finance Python programs
JAX-based neural network library
A lightweight vision library for performing large object detection
Build portable, production-ready MLOps pipelines
Toolkit for conversational AI
Fast image augmentation library and an easy-to-use wrapper
Data manipulation and transformation for audio signal processing
A general fine-tuning kit geared toward image/video/audio diffusion
AIMET is a library that provides advanced quantization and compression
Hub of ready-to-use datasets for ML models
A unified, comprehensive and efficient recommendation library
A curated list of data mining papers about fraud detection
NVIDIA Federated Learning Application Runtime Environment
A lightweight library for PyTorch training tools and utilities
Clean, Robust, and Unified PyTorch implementation
Optax is a gradient processing and optimization library for JAX
Test Suites for validating ML models & data
Graph Neural Network Library for PyTorch
Open-source Python framework for hybrid quantum-classical ml learning
machine learning tutorials (mainly in Python3)
Minimal and clean examples of machine learning algorithms
Probabilistic reasoning and statistical analysis in TensorFlow
Solve end to end problems using Llama model family