JAX-DFT

github.comsite

One-dimensional density functional theory (DFT) in JAX, with implementation of Kohn-Sham equations as regularizer: building prior knowledge into machine-learned physics.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   JaxDF github.com

   tool category #jax link ↗

   Framework for differentiable simulators with arbitrary discretizations.

   
2. ▲
   JAX, M.D. github.com

   tool category #jax link ↗

   Accelerated, differential molecular dynamics.

   
3. ▲
   FDTDX github.com

   tool category #jax link ↗

   Finite-Difference Time-Domain Electromagnetic Simulations in JAX

   
4. ▲
   dynamiqs github.com

   tool category #jax link ↗

   High-performance and differentiable simulations of quantum systems with JAX.

   
5. ▲
   JAX, M.D.: A Framework for Differentiable Physics. Samuel S. Schoenholz, Ekin D. Cubuk. NeurIPS 2020. arxiv.org

   site category #jax link ↗

   Introduces JAX, M.D., a differentiable physics library which includes simulation environments, interaction potentials, neural networks, and more.

   
6. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   

JAX › Models and Projects > JAX: “One-dimensional density functional theory (DFT) in JAX, with implementation of Kohn-Sham equations as regularizer: building prior knowledge into machine-learned physics.”