English
Related papers

Related papers: TorchDyn: A Neural Differential Equations Library

200 papers

This paper puts forward the vision of creating a library of neural-network-based models for power system simulations. Traditional numerical solvers struggle with the growing complexity of modern power systems, necessitating faster and more…

Systems and Control · Electrical Eng. & Systems 2025-02-11 Ioannis Karampinis , Petros Ellinas , Ignasi Ventura Nadal , Rahul Nellikkath , Spyros Chatzivasileiadis

We introduce a new family of deep neural network models. Instead of specifying a discrete sequence of hidden layers, we parameterize the derivative of the hidden state using a neural network. The output of the network is computed using a…

Machine Learning · Computer Science 2019-12-17 Ricky T. Q. Chen , Yulia Rubanova , Jesse Bettencourt , David Duvenaud

Deep Learning has emerged as one of the most significant innovations in machine learning. However, a notable limitation of this field lies in the ``black box" decision-making processes, which have led to skepticism within groups like…

Machine Learning · Computer Science 2025-03-06 Shi Li

In this work, we present a general purpose deep neural network package for representing energies, forces, dipole moments, and polarizabilities of atomistic systems. This so-called recursively embedded atom neural network model takes both…

Chemical Physics · Physics 2022-04-06 Yaolong Zhang , Junfan Xia , Bin Jiang

Continuous deep learning architectures have recently re-emerged as Neural Ordinary Differential Equations (Neural ODEs). This infinite-depth approach theoretically bridges the gap between deep learning and dynamical systems, offering a…

Machine Learning · Computer Science 2021-01-12 Stefano Massaroli , Michael Poli , Jinkyoo Park , Atsushi Yamashita , Hajime Asama

Three-dimensional (3D) point cloud analysis has become central to applications ranging from autonomous driving and robotics to forestry and ecological monitoring. Although numerous deep learning methods have been proposed for point cloud…

Computer Vision and Pattern Recognition · Computer Science 2026-04-14 Said Ohamouddou , Hanaa El Afia , Abdellatif El Afia , Raddouane Chiheb

The article presents the torchosr package - a Python package compatible with PyTorch library - offering tools and methods dedicated to Open Set Recognition in Deep Neural Networks. The package offers two state-of-the-art methods in the…

Machine Learning · Computer Science 2024-02-12 Joanna Komorniczak , Pawel Ksieniewicz

Automatic differentiation frameworks are optimized for exactly one thing: computing the average mini-batch gradient. Yet, other quantities such as the variance of the mini-batch gradients or many approximations to the Hessian can, in…

Machine Learning · Computer Science 2020-02-18 Felix Dangel , Frederik Kunstner , Philipp Hennig

In this paper, we introduce MCTensor, a library based on PyTorch for providing general-purpose and high-precision arithmetic for DL training. MCTensor is used in the same way as PyTorch Tensor: we implement multiple basic, matrix-level…

Machine Learning · Computer Science 2022-08-31 Tao Yu , Wentao Guo , Jianan Canal Li , Tiancheng Yuan , Christopher De Sa

We present "torchGDM", a numerical framework for nano-optical simulations based on the Green's Dyadic Method (GDM). This toolkit combines a hybrid approach, allowing for both fully discretized nano-structures and structures approximated by…

Physics-informed learning has shown to have a better generalization than learning without physical priors. However, training physics-informed deep neural networks requires some aspect of physical simulations to be written in a…

Machine Learning · Computer Science 2020-10-06 Muhammad F. Kasim , Sam M. Vinko

We present PyTorch Frame, a PyTorch-based framework for deep learning over multi-modal tabular data. PyTorch Frame makes tabular deep learning easy by providing a PyTorch-based data structure to handle complex tabular data, introducing a…

Machine Learning · Computer Science 2024-12-17 Weihua Hu , Yiwen Yuan , Zecheng Zhang , Akihiro Nitta , Kaidi Cao , Vid Kocijan , Jinu Sunil , Jure Leskovec , Matthias Fey

Neural networks are increasingly deployed in scientific, safety critical, and mission critical pipelines, yet verification and analysis are often performed outside the programming environment that defines and runs the model. This creates a…

Mathematical Software · Computer Science 2026-05-26 Robert Joseph George , Jennifer Cruden , Will Adkisson , Xiangru Zhong , Huan Zhang , Anima Anandkumar

Unsupervised learning of disentangled representations is an open problem in machine learning. The Disentanglement-PyTorch library is developed to facilitate research, implementation, and testing of new variational algorithms. In this…

Machine Learning · Computer Science 2019-12-12 Amir H. Abdi , Purang Abolmaesumi , Sidney Fels

In this paper, we introduce McTorch, a manifold optimization library for deep learning that extends PyTorch. It aims to lower the barrier for users wishing to use manifold constraints in deep learning applications, i.e., when the parameters…

Machine Learning · Statistics 2018-10-05 Mayank Meghwanshi , Pratik Jawanpuria , Anoop Kunchukuttan , Hiroyuki Kasai , Bamdev Mishra

We introduce gvnn, a neural network library in Torch aimed towards bridging the gap between classic geometric computer vision and deep learning. Inspired by the recent success of Spatial Transformer Networks, we propose several new layers…

Computer Vision and Pattern Recognition · Computer Science 2016-08-15 Ankur Handa , Michael Bloesch , Viorica Patraucean , Simon Stent , John McCormac , Andrew Davison

We present DeepAL, a Python library that implements several common strategies for active learning, with a particular emphasis on deep active learning. DeepAL provides a simple and unified framework based on PyTorch that allows users to…

Machine Learning · Computer Science 2021-12-01 Kuan-Hao Huang

In recent years the study of deep learning for solving differential equations has grown substantially. The use of physics-informed neural networks (PINNs) and deep operator networks (DeepONets) have emerged as two of the most useful…

Machine Learning · Computer Science 2025-08-27 Jason Matthews , Alex Bihlo

Deep learning has been highly successful in some applications. Nevertheless, its use for solving partial differential equations (PDEs) has only been of recent interest with current state-of-the-art machine learning libraries, e.g.,…

Machine Learning · Computer Science 2024-10-22 Hamid El Bahja , Jan Christian Hauffen , Peter Jung , Bubacarr Bah , Issa Karambal

Continual Learning is an important and challenging problem in machine learning, where models must adapt to a continuous stream of new data without forgetting previously acquired knowledge. While existing frameworks are built on PyTorch, the…

Machine Learning · Computer Science 2023-04-24 Nikolaos Dimitriadis , Francois Fleuret , Pascal Frossard