English
Related papers

Related papers: TorchDyn: A Neural Differential Equations Library

200 papers

We introduce Torch-Points3D, an open-source framework designed to facilitate the use of deep networks on3D data. Its modular design, efficient implementation, and user-friendly interfaces make it a relevant tool for research and…

Computer Vision and Pattern Recognition · Computer Science 2020-10-12 Thomas Chaton , Nicolas Chaulet , Sofiane Horache , Loic Landrieu

We introduce PyTorch Geometric, a library for deep learning on irregularly structured input data such as graphs, point clouds and manifolds, built upon PyTorch. In addition to general graph data structures and processing methods, it…

Machine Learning · Computer Science 2019-04-26 Matthias Fey , Jan Eric Lenssen

This work is a rigorous development of a complete and general-purpose deep learning framework from the ground up. The fundamental components of deep learning - automatic differentiation and gradient methods of optimizing multivariable…

Mathematical Software · Computer Science 2020-11-18 Andrei Nicolae

Physics informed neural networks (PINNs) have emerged as a powerful tool to provide robust and accurate approximations of solutions to partial differential equations (PDEs). However, PINNs face serious difficulties and challenges when…

Machine Learning · Computer Science 2023-07-11 Rajat Arora

We present PyTorch Geometric Temporal a deep learning framework combining state-of-the-art machine learning algorithms for neural spatiotemporal signal processing. The main goal of the library is to make temporal geometric deep learning…

This paper presents the design, implementation, and evaluation of the PyTorch distributed data parallel module. PyTorch is a widely-adopted scientific computing package used in deep learning research and applications. Recent advances in…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-06-30 Shen Li , Yanli Zhao , Rohan Varma , Omkar Salpekar , Pieter Noordhuis , Teng Li , Adam Paszke , Jeff Smith , Brian Vaughan , Pritam Damania , Soumith Chintala

This work presents Kornia -- an open source computer vision library which consists of a set of differentiable routines and modules to solve generic computer vision problems. The package uses PyTorch as its main backend both for efficiency…

Computer Vision and Pattern Recognition · Computer Science 2019-10-10 Edgar Riba , Dmytro Mishkin , Daniel Ponsa , Ethan Rublee , Gary Bradski

Solving complex computer vision tasks by deep learning techniques relies on large amounts of (supervised) image data, typically unavailable in industrial environments. The lack of training data starts to impede the successful transfer of…

Computer Vision and Pattern Recognition · Computer Science 2020-10-23 Christoph Heindl , Lukas Brunner , Sebastian Zambal , Josef Scharinger

Despite significant progress of applying deep learning methods to the field of content-based image retrieval, there has not been a software library that covers these methods in a unified manner. In order to fill this gap, we introduce…

Information Retrieval · Computer Science 2020-08-06 Benyi Hu , Ren-Jie Song , Xiu-Shen Wei , Yazhou Yao , Xian-Sheng Hua , Yuehu Liu

We revisit the original approach of using deep learning and neural networks to solve differential equations by incorporating the knowledge of the equation. This is done by adding a dedicated term to the loss function during the optimization…

Machine Learning · Computer Science 2023-04-05 Hubert Baty , Leo Baty

The research in Artificial Intelligence methods with potential applications in science has become an essential task in the scientific community last years. Physics Informed Neural Networks (PINNs) is one of this methods and represent a…

Computational Physics · Physics 2023-07-24 Luis Medrano Navarro , Luis Martín Moreno , Sergio G Rodrigo

In recent years, deep learning has witnessed its blossom in the field of Electrocardiography (ECG) processing, outperforming traditional signal processing methods in various tasks, for example, classification, QRS detection, wave…

Machine Learning · Computer Science 2022-04-12 Wen Hao , Kang Jingsu

This work presents Kornia, an open source computer vision library built upon a set of differentiable routines and modules that aims to solve generic computer vision problems. The package uses PyTorch as its main backend, not only for…

Computer Vision and Pattern Recognition · Computer Science 2020-09-23 E. Riba , D. Mishkin , J. Shi , D. Ponsa , F. Moreno-Noguer , G. Bradski

Deep learning has become a pivotal technology in fields such as computer vision, scientific computing, and dynamical systems, significantly advancing these disciplines. However, neural Networks persistently face challenges related to…

Machine Learning · Computer Science 2025-10-14 Yongshuai Liu , Lianfang Wang , Kuilin Qin , Qinghua Zhang , Faqiang Wang , Li Cui , Jun Liu , Yuping Duan , Tieyong Zeng

We present TabMixNN, a flexible PyTorch-based deep learning framework that synthesizes classical mixed-effects modeling with modern neural network architectures for tabular data analysis. TabMixNN addresses the growing need for methods that…

Machine Learning · Computer Science 2026-01-01 Deniz Akdemir

Python has become the de-facto language for training deep neural networks, coupling a large suite of scientific computing libraries with efficient libraries for tensor computation such as PyTorch or TensorFlow. However, when models are used…

Machine Learning · Computer Science 2021-04-02 Zachary DeVito , Jason Ansel , Will Constable , Michael Suo , Ailing Zhang , Kim Hazelwood

Differentiable sparse linear algebra is foundational for scientific machine learning, yet PyTorch lacks a unified library for it: \texttt{torch.sparse} provides only low-level kernels and a non-differentiable, CPU-only \texttt{spsolve}, and…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-07 Mingyuan Chi , Shizheng Wen

This paper describes a deep-SDM framework, MALPOLON. Written in Python and built upon the PyTorch library, this framework aims to facilitate training and inferences of deep species distribution models (deep-SDM) and sharing for users with…

Machine Learning · Computer Science 2024-09-27 Theo Larcher , Lukas Picek , Benjamin Deneu , Titouan Lorieul , Maximilien Servajean , Alexis Joly

Deep learning, a branch of artificial intelligence, is a data-driven method that uses multiple layers of interconnected units or neurons to learn intricate patterns and representations directly from raw input data. Empowered by this…

Machine Learning · Computer Science 2025-07-28 Mohd Halim Mohd Noor , Ayokunle Olalekan Ige

When engineers train deep learning models, they are very much 'flying blind'. Commonly used methods for real-time training diagnostics, such as monitoring the train/test loss, are limited. Assessing a network's training process solely…

Machine Learning · Computer Science 2021-10-27 Frank Schneider , Felix Dangel , Philipp Hennig