=====================================
Code for Neuromechanical Autoencoders
=====================================

The main folder has the launch scripts for the experiments. The scripts rely on config files in the config/ directory.

shape_constructors/ construct the B-spline representation of the geometry for the experiments.
varmintv2/ is the codebase of the solver we wrote for nonlinear elasticity.

The main requirements for the codebase is mpi4py, JAX, numpy/scipy/matplotlib, quadpy, absl-py, Haiku, optax, ml_collections
