AI Toolkit for Healthcare Imaging
The Unified Machine Learning Framework
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
Self-supervised visual learning using momentum contrast in PyTorch
Machine learning metrics for distributed, scalable PyTorch application
SpikingJelly is an open-source deep learning framework
Machine learning metrics for distributed, scalable PyTorch application
An elegent pytorch implement of transformers
Graph Neural Network Library for PyTorch
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
12 Weeks, 24 Lessons, AI for All
Simulation of spiking neural networks (SNNs) using PyTorch
Accessible large language models via k-bit quantization for PyTorch
A PyTorch library for implementing flow matching algorithms
From Images to High-Fidelity 3D Assets
Training PyTorch models with differential privacy
A cross-platform Python library for differentiable programming
A trainable PyTorch reproduction of AlphaFold 3