Getting Started

Prerequisites

• Programming Language: Python

• Libraries: jax, jax_tqdm, matplotlib

Installation

Using PyPi:

pip install jaxincell

Build from source:

git clone https://github.com/uwplasma/JAX-in-Cell
cd JAX-in-Cell
pip install -r /path/to/requirements.txt
pip install -e .

Usage

jaxincell

To run using an input TOML file:

jaxincell example_input.toml

Run example script:

python example_script.py

Testing

pytest .