C++ library for high performance inference on NVIDIA GPUs
Open standard for machine learning interoperability
Deep learning optimization library making distributed training easy
Jittor is a high-performance deep learning framework
An implementation of a deep learning recommendation model (DLRM)
Decentralized deep learning in PyTorch. Built to train models
Time series Timeseries Deep Learning Machine Learning Pytorch fastai
Graph Neural Network Library for PyTorch
A machine learning software for extracting information
A simple forecasting package
Probabilistic time series modeling in Python
PArallel Distributed Deep LEarning: Machine Learning Framework
Unity machine learning agents toolkit
Ongoing research training transformer models at scale
Deep learning optimization library: makes distributed training easy
A ranked list of awesome machine learning Python libraries
ktrain is a Python library that makes deep learning AI more accessible
Industrial-strength Natural Language Processing (NLP)
ImageBind One Embedding Space to Bind Them All
An industrial grade federated learning framework
Open deep learning compiler stack for cpu, gpu, etc.
PyTorch code and models for VJEPA2 self-supervised learning from video
AutoGluon: AutoML for Image, Text, and Tabular Data
Python package built to ease deep learning on graph
Best practices on recommendation systems