Enable jax.Hyperelastic(..., parallel=True)
with jax.pmap
on quadrature-points axis
#649
Job | Run time |
---|---|
2m 52s | |
3m 2s | |
5m 54s |