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
Comgra: A Tool for Analyzing and Debugging Neural Networks
Florian Dietz, Sophie Fellenz, Dietrich Klakow, Marius Kloft
2024-08-01
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
pyvene: A Library for Understanding and Improving PyTorch Models via Interventions
Zhengxuan Wu, Atticus Geiger, Aryaman Arora, Jing Huang +4
2024-03-13
Computation and Language · Computer Science
PyText: A Seamless Path from NLP research to production
Ahmed Aly, Kushal Lakhotia, Shicong Zhao, Mrinal Mohit +6
2018-12-21
Computation and Language · Computer Science
Using Captum to Explain Generative Language Models
Vivek Miglani, Aobo Yang, Aram H. Markosyan, Diego Garcia-Olano +1
2023-12-12
Earth and Planetary Astrophysics · Physics
PyLightcurve-torch: a transit modelling package for deep learning applications in PyTorch
Mario Morvan, Angelos Tsiaras, Nikolaos Nikolaou, Ingo P. Waldmann
2021-03-24
Information Retrieval · Computer Science
ir_explain: a Python Library of Explainable IR Methods
Sourav Saha, Harsh Agarwal, V Venktesh, Avishek Anand +3
2025-05-13
Computation and Language · Computer Science
Extending Logic Explained Networks to Text Classification
Rishabh Jain, Gabriele Ciravegna, Pietro Barbiero, Francesco Giannini +2
2023-09-28
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
depyf: Open the Opaque Box of PyTorch Compiler for Machine Learning Researchers
Kaichao You, Runsheng Bai, Meng Cao, Jianmin Wang +2
2024-03-22
Computer Vision and Pattern Recognition · Computer Science
Kaolin: A PyTorch Library for Accelerating 3D Deep Learning Research
Krishna Murthy Jatavallabhula, Edward Smith, Jean-Francois Lafleche, Clement Fuji Tsang +5
2019-11-14
Machine Learning · Computer Science
torchtune: PyTorch native post-training library
Mark Obozov, Maxime Griot, Joseph Cummings, Evan Smothers +7
2026-05-21
Machine Learning · Computer Science
DeepLog: A Software Framework for Modular Neurosymbolic AI
Robin Manhaeve, Stefano Colamonaco, Vincent Derkinderen, Rik Adriaensen +3
2026-05-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