Library for PDE optimization and control with gradient-based methods and reinforcement learning
git clone https://github.com/acoh64/pde-opt.git
cd pde-opt
conda create -y -n pde-opt-env python=3.10
conda activate pde-opt-env
pip install -e .
pip install -U "jax[cuda12]==0.6.2" "jaxlib==0.6.2"