Tensors and Dynamic neural networks in Python with strong GPU acceleration

Ultralytics YOLO with Additional Knowledge Distillation Capability

An unofficial PyTorch implementation of a federated learning algorithm, FedAvg.

[AISTATS 2024] On-Demand Federated Learning for Arbitrary Target Class Distributions

Open deep learning compiler stack for cpu, gpu and specialized accelerators