Installation

Installing via PyPI

We recommend using a Conda environment (or another virtual environment) when installing GRaTeR-JAX to ensure that dependencies are properly isolated and compatible.

For a standard installation, create a new environment with a supported Python version and run:

pip install grater-jax

Make sure JAX is installed with the appropriate backend for your hardware:

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

Compatibility

GRaTeR-JAX is compatible with Python versions >=3.9 and <3.13, and with GPU hardware supported by JAX. See the JAX installation documentation for supported backends and platforms.

A full list of package dependencies is available in pyproject.toml.