Libraries

When running in colab do

!pip install zuko

Installation of the environment:

uv venv zuko_tester_env

Activate it

source zuko_tester_env/bin/activate

Install scientific + Jupyter stack

uv pip install numpy scipy pandas matplotlib seaborn tqdm scikit-learn jupyterlab ipykernel

Install PyTorch and Zuko

uv pip install torch torchvision torchaudio zuko