Pytorch domain library for recommendation systems
BSD-3-CLAUSE License
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iter...
Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reprod...
Experiments of Pytorch SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
Enabling PyTorch on XLA Devices (e.g. Google TPU)
OCaml bindings for PyTorch
Torch Distributed Experimental
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
PyTorch Extension Library of Optimized Scatter Operations
The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR...
A powerful and flexible machine learning platform for drug discovery
Exploration of Self-Semi-Supervised learning for handling unlabeled data
A native PyTorch Library for large model training
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT