requirements.txt

tensorflow
jaxlib
jax
jaxrk
nest-asyncio
pyfluidsynth==1.3.0
clu==0.0.7
orbax==0.0.2
jax[cuda]==0.3.15 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html