JAX

swMATH32996MaRDI QIDQ44707FDOQ44707

Author name not available (Why is that?)

Source code repository: https://github.com/google/jax

Cited In (41)

• Statistical Finite Elements via Langevin Dynamics
• Data-driven tissue mechanics with polyconvex neural ordinary differential equations
• Learning probabilistic termination proofs
• Scalable uncertainty quantification for deep operator networks using randomized priors
• Explicitly antisymmetrized neural network layers for variational Monte Carlo simulation
• Learning stochastic dynamics with statistics-informed neural network
• A Simple and Efficient Tensor Calculus for Machine Learning
• Multilevel Fine-Tuning: Closing Generalization Gaps in Approximation of Solution Maps under a Limited Budget for Training Data
• Wide neural networks of any depth evolve as linear models under gradient descent ^*
• Deep neural networks and adaptive quadrature for solving variational problems
• WENO3-NN: a maximum-order three-point data-driven weighted essentially non-oscillatory scheme
• Dynamic calibration of differential equations using machine learning, with application to turbulence models
• Optimal Energy Shaping via Neural Approximators
• mad-GP: automatic differentiation of Gaussian processes for molecules and materials
• Optimal Transport for Parameter Identification of Chaotic Dynamics via Invariant Measures
• Improved architectures and training algorithms for deep operator networks
• DisCoPy: Monoidal Categories in Python
• JAX, M.D. A framework for differentiable physics*
• Discovering faster matrix multiplication algorithms with reinforcement learning
• A cornucopia of \(\mathrm{AdS}_5\) vacua
• Automated Reinforcement Learning (AutoRL): A Survey and Open Problems
• Title not available (Why is that?)
• Cryptanalytic extraction of neural network models
• Title not available (Why is that?)
• Autodifferentiable Ensemble Kalman Filters
• Extension of dynamic mode decomposition for dynamic systems with incomplete information based on t-model of optimal prediction
• Long-time integration of parametric evolution equations with physics-informed DeepONets
• Title not available (Why is that?)
• Exact imposition of boundary conditions with distance functions in physics-informed deep neural networks
• On the eigenvector bias of Fourier feature networks: from regression to solving multi-scale PDEs with physics-informed neural networks
• Optimum design of nonlinear structures via deep neural network-based parameterization framework
• Majorization-minimization-based Levenberg-Marquardt method for constrained nonlinear least squares
• Halting time is predictable for large models: a universality property and average-case analysis
• The hierarchical parity model
• Moduli-dependent Calabi-Yau and SU(3)-structure metrics from machine learning
• Least-squares solutions of boundary-value problems in hybrid systems
• Imitation learning of stabilizing policies for nonlinear systems
• Emulation of cardiac mechanics using graph neural networks
• rigidPY: rigidity analysis in Python
• Simulation of the 3D hyperelastic behavior of ventricular myocardium using a finite-element based neural-network approach
• A Model-Constrained Tangent Slope Learning Approach for Dynamical Systems

This page was built for software: JAX