jax-am
jax-am copied to clipboard
tianjuxue
Additive manufacturing simulation with JAX.
I am trying to run the demo, but failed with: ImportError: cannot import name 'config' from 'jax.config' (.../envs/jax-am/lib/python3.9/site-packages/jax/config.py) It seems to be an error from mismatched (i.e., newer) jax version,...
Hello, I am attempting to solve a 2D plane strain boundary value problem using a linear elastic material with Young's modulus that varies spatially, expressed as E = E(x,y) or...
It seems there is come memory leakage somewhere. It can be clearly seen if the topopt example is run and you monitor the memory. It is steadily increasing. This results...
In `solver.py`, the following lines ``` tol = 1e-6 while res_val > tol: dofs = linear_incremental_solver(problem, res_vec, A_fn, dofs, precond, use_petsc) res_vec, A_fn = newton_update_helper(dofs) # test_jacobi_precond(problem, jacobi_preconditioner(problem, dofs), A_fn)...
- At the moment, for some BCs (especially in TopOpt), the petsc version converges but the jax_solve does not. - From our tests, we understood that this is due to...
This has to be fixed! We may need to keep short gradient checks for each function to detect where the leak is happening.
For now, we have used `device_put()` to put the global stiffness matrix into CPU. Need to create a transfer guard to ensure that this is indeed happening.
Dear @tianjuxue, I noticed a problematic use of assert statements in the FEM's solver.py code which could lead to failures. Here is the relevant code: ``` def jax_solve(problem, A_fn, b,...
