sailfish009 Goto Github PK

torch-quiver

PyTorch Library for Fast and Easy Distributed Graph Learning

torch-toolbox

[Active development]ToolBox to make using Pytorch much easier.

torch2trt

An easy to use PyTorch to TensorRT converter

torchani

Accurate Neural Network Potential on PyTorch

torchbearer

torchbearer: A model fitting library for PyTorch

torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

torchdrug

A powerful and flexible machine learning platform for drug discovery

torchdyn

A PyTorch based library for all things neural differential equations and implicit neural models.

torchfi

TorchFI is a fault injection framework build on top of PyTorch for research purposes.

torchfunc

PyTorch functions & utilities to make your deep learning life easier

torchinfo

View model summaries in PyTorch!

torchio

Tools for medical image processing in deep learning and PyTorch.

torchkge

TorchKGE: Knowledge Graph embedding in Python and PyTorch.

torchlayers

Shape inference for PyTorch (like Keras) & new layers

torchmd

End-To-End Molecular Dynamics (MD) Engine using PyTorch

torchmd-net

Neural network potentials based on graph neural networks and equivariant transformers

torchphysics

torchprof

PyTorch layer-by-layer model profiler

torchproteinlibrary

PyTorch library of layers acting on protein representations

torchprune

A repository for pytorch-based pruning and research.

torchrec

Pytorch domain library for recommendation systems

torchrs

Deep Learning in Rust with the PyTorch API

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

torchsemiseg

[CVPR 2021] Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

torchsort

Fast, differentiable sorting and ranking in PyTorch

torchsparse

torchssl

A PyTorch-based library for semi-supervised learning (NeurIPS'21)

torchstat

Model analyzer in PyTorch

torchsummaryx

torchsummaryX: Improved visualization tool of torchsummary

torchts

Time series forecasting with PyTorch