This repository provides unofficial, partial JAX-implementation for 'Physics-Informed Neural Networks for shell structures'.
By re-writing the original code, we could obtain substantial speed-ups.
Please consult with result_jax_x64.txt
and result_torch_x64.txt
for a performance comparison.
Training a neural network in double-precision: JAX_ENABLE_X64=True python main.py
.
Measuring relative python err_measure.py --method=lbfgs
Please consult with requirements.txt
.
- Fully clamped hyperbolic paraboloid