Usage

To run a simple case of JAX-in-Cell, call:

jaxincell

To use a TOML input file:

jaxincell example_input.toml

To run the example script:

python example_script.py