-
NUAA/XAC/FAI
- NanJing/XiAn
- http://liung.github.io
Stars
jax
4 repositories
Optax is a gradient processing and optimization library for JAX.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Interpolation and function approximation with JAX
Example problems in Physics informed neural network in JAX