Doom-based AI research platform for reinforcement learning
An open source implementation of CLIP
NVIDIA Federated Learning Application Runtime Environment
A PyTorch-based Speech Toolkit
A game theoretic approach to explain the output of ml models
Beta Machine Learning Toolkit
AIMET is a library that provides advanced quantization and compression
OpenMLDB is an open-source machine learning database
A fast library for AutoML and tuning
Graph Neural Network Library for PyTorch
Time series forecasting with PyTorch
Core ML tools contain supporting tools for Core ML model conversion
Uncover insights, surface problems, monitor, and fine tune your LLM
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
JAX-based neural network library
A collection of machine learning examples and tutorials
Solve end to end problems using Llama model family
Machine learning image inpainting task that removes watermarks
Personal notes from Wu Enda's machine learning course
A simple but complete full-attention transformer
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
AI Toolkit for Healthcare Imaging
Detecting silent model failure. NannyML estimates performance
Build cross-modal and multimodal applications on the cloud
A python library for easy manipulation and forecasting of time series