Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Example problems in Physics informed neural network in JAX

Simple implementation of Physics-Informed Neural Networks for the solution of Partial Differential Equations in JAX (using Equinox and Optax)

Solve forward and inverse problems related to partial differential equations using finite basis physics-informed neural networks (FBPINNs)

Blog posts on best practices for Julia development