Jacare

River routing in JAX.

Installation

Install with

pip install git+https://github.com/limamau/jacare.git

To enable JAX in the GPU, it may be necessary to manually install it inside an environment with CUDA and GPUs available through

pip install -U "jax[cuda12]"

Quick start

Build up the toy dataset with data/generate_dataset.py and play with the examples in

• independent catchments

• complete river routing