Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
Solve end to end problems using Llama model family
NVIDIA Federated Learning Application Runtime Environment
A machine learning library for detecting anomalies in signals
Training data (data labeling, annotation, workflow) for all data types
SpikingJelly is an open-source deep learning framework
A comprehensive quantitative trading system with AI-powered analysis
PyTorch version of Stable Baselines
Spatiotemporal Signal Processing with Neural Machine Learning Models
Data science on data without acquiring a copy
Python package built to ease deep learning on graph
MiniSom is a minimalistic implementation of the Self Organizing Maps
TFX is an end-to-end platform for deploying production ML pipelines
Pretrained (Language) Models for Probabilistic Time Series Forecasting
A high performance implementation of HDBSCAN clustering
SkyPilot: Run AI and batch jobs on any infra
Open-source tool designed to enhance the efficiency of workloads
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Advanced AI Explainability for computer vision
A library for accelerating Transformer models on NVIDIA GPUs
End-to-End Library for Continual Learning based on PyTorch