Publications on JAX
Introduction
Applications
Tools
Research Groups
Workshops
Publications
   List Publications
   Advanced Search
   Info
   Add Publications
My Account
About
Impress


Nina Baldy, Marmaduke Woodman, Viktor K. Jirsa, Meysam Hashemi
Dynamic causal modelling in probabilistic programming languages
Article in Journal of the Royal Society Interface, The Royal Society, 2025
Application Area:
Probabilistic Programming
Tools:
JAX
Daniel Kelshaw, Luca Magri
Computing distances and means on manifolds with a metric-constrained Eikonal approach
Article in Proceedings of the Royal Society A: Mathematical, Physical and Engineering Sciences, The Royal Society, 2025
Application Area:
Geometry
Tools:
JAX
Deniz A. Bezgin, Aaron B. Buhendwa, Nikolaus A. Adams
JAX-Fluids 2.0: Towards HPC for differentiable CFD of compressible two-phase flows
Article in Computer Physics Communications, 2025
Application Area:
Computational Fluid Dynamics
Tools:
JAX
Adam S. Abbott, Boyi Z. Abbott, Justin M. Turney, Henry F. Schaefer
Arbitrary-Order Derivatives of Quantum Chemical Methods via Automatic Differentiation
Article in The Journal of Physical Chemistry Letters, American Chemical Society (ACS), 2021
Application Area:
Chemistry
Tools:
JAX
N. McGreivy, S. R. Hudson, C. Zhu
Optimized finite-build stellarator coils using automatic differentiation
Article in Nuclear Fusion, IOP Publishing, 2021
Application Area:
Nuclear Fusion
Tools:
JAX
Samuel Schoenholz, Ekin Dogus Cubuk
JAX MD: A Framework for Differentiable Physics
Conference proceeding, Advances in Neural Information Processing Systems, Curran Associates, Inc., 2020
Application Area:
Molecular Dynamics
Tools:
JAX

back
  

Contact:
autodiff.org
Username:
Password:
(lost password)