Equinox: neural networks in JAX via callable PyTrees and filtered transformations

30 October 2021

Papers citing "Equinox: neural networks in JAX via callable PyTrees and filtered transformations"

7 / 57 papers shown

Title
Hybridizing Physics and Neural ODEs for Predicting Plasma Inductance Dynamics in Tokamak Fusion Reactors Allen M. Wang Darren T. Garnier Cristina Rea AI4CE 17 3 0 30 Oct 2023
Enhancing Low-Order Discontinuous Galerkin Methods with Neural Ordinary Differential Equations for Compressible Navier--Stokes Equations Shinhoo Kang Emil M. Constantinescu AI4CE 27 0 0 29 Oct 2023
SynJax: Structured Probability Distributions for JAX Miloš Stanojević Laurent Sartran SyDa 18 4 0 07 Aug 2023
Pseudo-rigid body networks: learning interpretable deformable object dynamics from partial observations Shamil Mamedov A. R. Geist Jan Swevers Sebastian Trimpe AI4CE 26 2 0 16 Jul 2023
Data-driven multiscale modeling for correcting dynamical systems Karl Otness L. Zanna Joan Bruna AI4CE 39 6 0 24 Mar 2023
GECCO: Geometrically-Conditioned Point Diffusion Models M. Tyszkiewicz Pascal Fua Eduard Trulls DiffM 26 21 0 10 Mar 2023
Noncommutative $C^$ -algebra Net: Learning Neural Networks with Powerful Product Structure in $C^$ -algebra Ryuichiro Hataya Yuka Hashimoto 52 4 0 26 Jan 2023