tianjuxue / jax-am

Additive manufacturing simulation with JAX.
https://jax-am.readthedocs.io/en/latest/
GNU General Public License v3.0
273 stars 56 forks source link

Topology Optimization RL with Keras #14

Closed kbali1297 closed 1 year ago

kbali1297 commented 1 year ago

Working example for Topology Optimization over 6X6 element matrix using RL with Keras. Run optimizer.py for training the RL agent. Currently doesn't give optimal topologies and still under debugging phase.