Machine Learning · Computer Science
Torch.fx: Practical Program Capture and Transformation for Deep Learning in Python
James K. Reed, Zachary DeVito, Horace He, Ansley Ussery +1
2022-03-08
Computer Vision and Pattern Recognition · Computer Science
PyTorchVideo: A Deep Learning Library for Video Understanding
Haoqi Fan, Tullie Murrell, Heng Wang, Kalyan Vasudev Alwala +12
2021-11-19
Machine Learning · Computer Science
Torchmeta: A Meta-Learning library for PyTorch
Tristan Deleu, Tobias Würfl, Mandana Samiei, Joseph Paul Cohen +1
2019-09-17
Machine Learning · Computer Science
PyTorch: An Imperative Style, High-Performance Deep Learning Library
Adam Paszke, Sam Gross, Francisco Massa, Adam Lerer +17
2019-12-05
Machine Learning · Computer Science
TorchBench: Benchmarking PyTorch with High API Surface Coverage
Yueming Hao, Xu Zhao, Bin Bao, David Berard +3
2023-06-27
Machine Learning · Computer Science
Using Python for Model Inference in Deep Learning
Zachary DeVito, Jason Ansel, Will Constable, Michael Suo +2
2021-04-02
Machine Learning · Computer Science
torchgfn: A PyTorch GFlowNet library
Joseph D. Viviano, Omar G. Younis, Sanghyeok Choi, Victor Schmidt +2
2026-03-23
Machine Learning · Computer Science
torchmil: A PyTorch-based library for deep Multiple Instance Learning
Francisco M. Castro-Macías, Francisco J. Sáez-Maldonado, Pablo Morales-Álvarez, Rafael Molina
2025-09-11
Machine Learning · Computer Science
PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning
Weihua Hu, Yiwen Yuan, Zecheng Zhang, Akihiro Nitta +5
2024-12-17
Machine Learning · Computer Science
MCTensor: A High-Precision Deep Learning Library with Multi-Component Floating-Point
Tao Yu, Wentao Guo, Jianan Canal Li, Tiancheng Yuan +1
2022-08-31
Machine Learning · Computer Science
torchtune: PyTorch native post-training library
Mark Obozov, Maxime Griot, Joseph Cummings, Evan Smothers +7
2026-05-21
Computer Vision and Pattern Recognition · Computer Science
TerraTorch: The Geospatial Foundation Models Toolkit
Carlos Gomes, Benedikt Blumenstiel, Joao Lucas de Sousa Almeida, Pedro Henrique de Oliveira +6
2025-03-27
Machine Learning · Computer Science
Amazon SageMaker Model Parallelism: A General and Flexible Framework for Large Model Training
Can Karakus, Rahul Huilgol, Fei Wu, Anirudh Subramanian +6
2021-11-12
Information Retrieval · Computer Science
Lightning IR: Straightforward Fine-tuning and Inference of Transformer-based Language Models for Information Retrieval
Ferdinand Schlatt, Maik Fröbe, Matthias Hagen
2025-04-16
Image and Video Processing · Electrical Eng. & Systems
TorchXRayVision: A library of chest X-ray datasets and models
Joseph Paul Cohen, Joseph D. Viviano, Paul Bertin, Paul Morrison +7
2021-11-02
Machine Learning · Computer Science
TorchSurv: A Lightweight Package for Deep Survival Analysis
Mélodie Monod, Peter Krusche, Qian Cao, Berkman Sahiner +3
2024-04-18
Machine Learning · Computer Science
TorchScale: Transformers at Scale
Shuming Ma, Hongyu Wang, Shaohan Huang, Wenhui Wang +7
2022-11-24
Mathematical Software · Computer Science
Implementation of tangent linear and adjoint models for neural networks based on a compiler library tool
Sa Xiao, Hao Jing, Honglu Sun, Haoyu Li
2026-03-19
Programming Languages · Computer Science
When Language Model Meets Private Library
Daoguang Zan, Bei Chen, Zeqi Lin, Bei Guan +2
2022-11-01
Machine Learning · Computer Science
TorchDyn: A Neural Differential Equations Library
Michael Poli, Stefano Massaroli, Atsushi Yamashita, Hajime Asama +1
2020-09-22
Computer Vision and Pattern Recognition · Computer Science
DORAEMON: A Unified Library for Visual Object Modeling and Representation Learning at Scale
Ke Du, Yimin Peng, Chao Gao, Fan Zhou +1
2025-11-07