State-of-the-art 2D and 3D Face Analysis Project
A scikit-learn compatible neural network library that wraps PyTorch
Data manipulation and transformation for audio signal processing
On-device AI across mobile, embedded and edge for PyTorch
A lightweight library for PyTorch training tools and utilities
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
AI Toolkit for Healthcare Imaging
The Unified Machine Learning Framework
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
PyTorch implementation of JiT
Official inference repo for FLUX.1 models
Advanced evolutionary computation library built on top of PyTorch
Run PyTorch LLMs locally on servers, desktop and mobile
Machine learning metrics for distributed, scalable PyTorch application
SpikingJelly is an open-source deep learning framework
Graph Neural Network Library for PyTorch
Machine learning metrics for distributed, scalable PyTorch application
An elegent pytorch implement of transformers
Scalable generative AI framework built for researchers and developers
Tensor Learning in Python
A modular, primitive-first, python-first PyTorch library
Federated Learning (FL) experiment simulation in Python
Simulation framework for accelerating research
Algorithms for outlier, adversarial and drift detection
ONNX Runtime: cross-platform, high performance ML inferencing