diffmpm icon indicating copy to clipboard operation
diffmpm copied to clipboard
verified publisher icon geoelements

Update loops to use jax calls and JIT some functions

Open chahak13 opened this issue 2 years ago • 1 comments

chahak13 avatar Jul 19 '23 05:07 chahak13

image

Ran optimization with 2D Uniaxial stress example. Given stress, estimated Young's Modulus. This is with the stateless implementation. This is possible because https://github.com/geoelements/diffmpm/pull/27/commits/1034d4f03c2d9e72277137befc3a58333047875a changes all for loops to scans so that bidirectional differentiation can work.

@kks32 ^

chahak13 avatar Aug 17 '23 17:08 chahak13