## Installation

We recommend using Conda to create a fresh environment in which to run GRaTer-JAX, to ensure all dependencies are met:

```bash
conda env create -f environment.yml
conda activate grater-jax-env   # or your env name
pip install -e .

To install GRaTeR-JAX and its dependencies, create a new Conda environment with Python and run:

pip install grater-jax

Make sure you have JAX installed with the correct backend for your hardware:

pip install --upgrade "jax[cpu]"  # or "jax[cuda]" for GPU

NOTE: For Millar-Blanchaer Lab developers – you need to install with Jax 0.4.34 and Cuda 12.2 if using Mueller.

Compatibility

GRaTer-JAX is compatible with Python >=3.9 and <3.13, and any GPU processor supported by JAX (see the JAX install docs for a list). A full list of dependencies is available in pyproject.toml