torch-dist-utils

Utilities for PyTorch distributed

APACHE-2.0 License

Stars

Ecosystems: Python

Statistics for this project are still being loaded, please check back later.

An implementation of a deep learning recommendation model (DLRM)

My best practice of training large dataset using PyTorch.

Several simple examples for popular neural network toolkits calling custom CUDA operators.

SimCLR pytorch implementation using DistributedDataParallel.

NumPy实现类PyTorch的动态计算图和神经网络框架(MLP, CNN, RNN, Transformer)

An official implementation of "An Efficient Combinatorial Optimization Model Using Learning-to-Ra...

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

Build on top of pytorch to fuel productivity

The PyTorch Implementation based on YOLOv4 of the paper: "Complex-YOLO: Real-time 3D Object Detec...

gpu tester detects broken and slow gpus in a cluster