Automated deep learning algorithms implemented in PyTorch.
MIT License
DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbon...
NAS-Bench-201 API and Instruction
PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"
OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning
This project reproduces the book Dive Into Deep Learning (https://d2l.ai/), adapting the code fro...
A collection of computer vision pre-trained models.
Multi-Task Deep Neural Networks for Natural Language Understanding
Official Implementation of 'Fast AutoAugment' in PyTorch.
A PyTorch toolbox for domain generalization, domain adaptation and semi-supervised learning.
Differentiable architecture search for convolutional and recurrent networks
An autoML framework & toolkit for machine learning on graphs.
OTOv1-v3, NeurIPS, ICLR, TMLR, DNN Training, Compression, Structured Pruning, Erasing Operators, ...
Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral)
[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convoluti...
Automatic architecture search and hyperparameter optimization for PyTorch