PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Library for training machine learning models with privacy for data
TimeGPT-1: production ready pre-trained Time Series Foundation Model
Hub of ready-to-use datasets for ML models
Tool for visualizing and tracking your machine learning experiments
Plugin to integrate Learning to Rank
Relax! Flux is the ML library that doesn't make you tensor
Jittor is a high-performance deep learning framework
Collection of CVPR 2025 papers and open source projects
High-level, high-performance dynamic language for technical computing
A general-purpose probabilistic programming system
The Operator Splitting QP Solver
Deep Learning API and Server in C++14 support for Caffe, PyTorch
Please do not feed the models
High quality, fast, modular reference implementation of SSD in PyTorch
Implementation of DeepLabCut
Build MLOps Pipelines in Minutes
Solve end to end problems using Llama model family
Elyra extends JupyterLab with an AI centric approach
Resources for deep learning with satellite & aerial imagery
Optax is a gradient processing and optimization library for JAX
Python toolbox to create adversarial examples
Machine Learning automation and tracking
A Flexible and Powerful Parameter Server for large-scale ML
Powering Amazon custom machine learning chips