From 286ed5153494c8acab4e40e1bda59fe8b0f32865 Mon Sep 17 00:00:00 2001 From: Gnome Ann <> Date: Fri, 19 Nov 2021 18:20:02 -0500 Subject: [PATCH] Add a requirements.txt for TPU backend --- requirements_mtj.txt | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 requirements_mtj.txt diff --git a/requirements_mtj.txt b/requirements_mtj.txt new file mode 100644 index 00000000..d6a70923 --- /dev/null +++ b/requirements_mtj.txt @@ -0,0 +1,13 @@ +numpy +tqdm +requests +optax +dm-haiku >= 0.0.5, <= 0.0.9 +ray[default] +jax == 0.2.12 +transformers +progressbar2 +git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck +flask +Flask-SocketIO +flask-cloudflared >= 0.0.5