A library for scientific machine learning & physics-informed learning
Petastorm library enables single machine or distributed training
oneAPI Deep Neural Network Library (oneDNN)
SAPIEN Manipulation Skill Framework
RL research on Android devices
Create UIs for your machine learning model in Python in 3 minutes
Advanced evolutionary computation library built on top of PyTorch
ProtoMotions is a GPU-accelerated simulation and learning framework
Federated Learning (FL) experiment simulation in Python
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Train any agents simply by 'talking'
CV, NLP, LLM project applications, and advanced engineering deployment
End-to-End Library for Continual Learning based on PyTorch
A GPU-accelerated library containing highly optimized building blocks
Graph Neural Network Library for PyTorch
Port of OpenAI's Whisper model in C/C++
Enterprise-grade (40m+ LOC) codebase intelligence
Feature engineering package with sklearn like functionality
Plain python implementations of basic machine learning algorithms
Implementation of DeepLabCut
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
C++-based high-performance parallel environment execution engine
Adversarial Robustness Toolbox (ART) - Python Library for ML security
Fast image augmentation library and an easy-to-use wrapper
An extensive node suite that enables ComfyUI to process 3D inputs