PyPI page
Author:
Hang Jung Ling
Summary:
A JAX framework for reproducible physics-informed neural network training.
Latest version:
0.2.0b5
Required dependencies:
colorlog
|
flax
|
h5py
|
hydra-core
|
jax
|
mat73
|
numpy
|
omegaconf
|
optax
|
orbax-checkpoint
|
pyyaml
|
rich
|
rootutils
|
scipy
|
tqdm
Optional dependencies:
jax
|
tensorboard
|
wandb
Downloads last day:
16
Downloads last week:
44
Downloads last month:
200
7-day smoothing