requirements.txt

tensorflow
nest-asyncio
pyfluidsynth==1.3.0
# downgrade jax
# TODO(iansimon): remove this once we can use the latest t5x
jax[cpu]==0.3.15 -f https://storage.googleapis.com/jax-releases/jax_releases.html
# pin CLU for python 3.7 compatibility
clu==0.0.7
# pin Orbax to use Checkpointer
orbax==0.0.2