English
Related papers

Related papers: JAX, M.D.: A Framework for Differentiable Physics

200 papers

Recent results on supercomputers show that beyond 65K cores, the efficiency of molecular dynamics simulations of interfacial systems decreases significantly. In this paper, we introduce a dynamic cutoff method (DCM) for interfacial systems…

Computational Physics · Physics 2017-01-23 Paul Springer , Ahmed E. Ismail , Paolo Bientinesi

We present a differentiable formulation of rigid-body contact dynamics for objects and robots represented as compositions of convex primitives. Existing optimization-based approaches simulating contact between convex primitives rely on a…

NetworkDynamics.jl is an easy-to-use and computationally efficient package for working with heterogeneous dynamical systems on complex networks, written in Julia, a high-level, high-performance, dynamic programming language. By combining…

Mathematical Software · Computer Science 2021-07-02 Michael Lindner , Lucas Lincoln , Fenja Drauschke , Julia Monika Koulen , Hans Würfel , Anton Plietzsch , Frank Hellmann

In resent years, the software ecosystem for numerical simulation still remains fragmented, with different algorithms and discretization methods often implemented in isolation, each with distinct data structures and programming conventions.…

Numerical Analysis · Mathematics 2026-03-10 Yangyang Zheng , Huayi Wei , Yunqing Huang , Chunyu Chen , Tian Tian , Hanbin Liu , Wenbin Wang , Liang He

Recent developments in many-body potential energy representation via deep learning have brought new hopes to addressing the accuracy-versus-efficiency dilemma in molecular simulations. Here we describe DeePMD-kit, a package written in…

Computational Physics · Physics 2018-05-23 Han Wang , Linfeng Zhang , Jiequn Han , Weinan E

Approximating partial differential equations for extensive industrial and scientific applications requires leveraging the power of modern high-performance computing. In large-scale parallel computations, the geometrical discretisation…

Numerical Analysis · Mathematics 2024-12-30 Pere A. Martorell , Santiago Badia

We extend JAX with the capability to automatically differentiate higher-order functions (functionals and operators). By representing functions as a generalization of arrays, we seamlessly use JAX's existing primitive system to implement…

Programming Languages · Computer Science 2024-01-30 Min Lin

Efficient accelerator modeling and particle tracking are key for the design and configuration of modern particle accelerators. In this work, we present JuTrack, a nested accelerator modeling package developed in the Julia programming…

Accelerator Physics · Physics 2024-12-30 Jinyu Wan , Helena Alamprese , Christian Ratcliff , Ji Qiang , Yue Hao

Molecular-dynamics algorithms for systems of particles interacting through discrete or "hard" potentials are fundamentally different to the methods for continuous or "soft" potential systems. Although many software packages have been…

Computational Physics · Physics 2013-09-30 M. N. Bannerman , R. Sargant , L. Lue

Optimization time integrators are effective at solving complex multi-physics problems including deformable solids with non-linear material models, contact with friction, strain limiting, etc. For challenging problems, Newton-type optimizers…

Computational Engineering, Finance, and Science · Computer Science 2026-01-01 José Antonio Fernández-Fernández , Fabian Löschner , Lukas Westhofen , Andreas Longva , Jan Bender

Simulation modeling of robots, objects, and environments is the backbone for all model-based control and learning. It is leveraged broadly across dynamic programming and model-predictive control, as well as data generation for imitation,…

Robotics · Computer Science 2022-01-19 Fei Liu , Mingen Li , Jingpei Lu , Entong Su , Michael C. Yip

In this work, we aim to teach robots to manipulate various thin-shell materials. Prior works studying thin-shell object manipulation mostly rely on heuristic policies or learn policies from real-world video demonstrations, and only focus on…

Robotics · Computer Science 2024-04-02 Yian Wang , Juntian Zheng , Zhehuan Chen , Zhou Xian , Gu Zhang , Chao Liu , Chuang Gan

Molecular Dynamics (MD) simulation is a powerful tool for understanding the dynamics and structure of matter. Since the resolution of MD is atomic-scale, achieving long time-scale simulations with femtosecond integration is very expensive.…

Machine Learning · Computer Science 2022-04-27 Zijie Li , Kazem Meidani , Prakarsh Yadav , Amir Barati Farimani

Molecular dynamics simulations are one of the methods in scientific computing that benefit from GPU acceleration. For those devices, SYCL is a promising API for writing portable codes. In this paper, we present the case study of "HAL's MD…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-06-07 Viktor Skoblin , Felix Höfling , Steffen Christgau

Molecular dynamics (MD) simulations have transformed our understanding of the nanoscale, driving breakthroughs in materials science, computational chemistry, and several other fields, including biophysics and drug design. Even on exascale…

Electronic structure simulation (ESS) has been used for decades to provide quantitative scientific insights on an atomistic scale, enabling advances in chemistry, biology, and materials science, among other disciplines. Following standard…

Machine Learning · Computer Science 2024-06-06 Hatem Helal , Andrew Fitzgibbon

We present a differentiable extension of the VEROS ocean model, enabling automatic differentiation through its dynamical core. We describe the key modifications required to make the model fully compatible with JAX autodifferentiation…

Machine Learning · Computer Science 2025-11-24 Etienne Meunier , Said Ouala , Hugo Frezat , Julien Le Sommer , Ronan Fablet

Spontaneous structural rearrangements play a central role in the organization and function of complex biomolecular systems. In principle, physics-based computer simulations like Molecular Dynamics (MD) enable us to investigate these…

Quantum Physics · Physics 2026-03-19 Danial Ghamari , Philipp Hauke , Roberto Covino , Pietro Faccioli

A new approach to simulating warm and hot dense matter that combines density functional theory based calculations of the electronic structure to classical molecular dynamics simulations with pair interaction potentials is presented. The new…

Plasma Physics · Physics 2015-06-22 C. E. Starrett , J. Daligault , D. Saumon

MADNESS (multiresolution adaptive numerical environment for scientific simulation) is a high-level software environment for solving integral and differential equations in many dimensions that uses adaptive and fast harmonic analysis methods…