{"library":"keras-nightly","install":[{"cmd":"pip install keras-nightly","imports":["import keras","from keras import layers"]},{"cmd":"pip install tensorflow","imports":[]},{"cmd":"pip install jax[cuda12_pip] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html","imports":[]},{"cmd":"pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121","imports":[]}]}