From c0b0af8d9de20a209b0640c6c74e3d6d1d3361f6 Mon Sep 17 00:00:00 2001 From: Chandra Devarakonda Date: Mon, 16 Oct 2023 23:00:44 +0000 Subject: [PATCH] Update praxis 1.2.0 requirements --- praxis/pip_package/requirements.txt | 4 ++-- requirements.in | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/praxis/pip_package/requirements.txt b/praxis/pip_package/requirements.txt index 30680d84..0eae5e75 100644 --- a/praxis/pip_package/requirements.txt +++ b/praxis/pip_package/requirements.txt @@ -104,7 +104,7 @@ executing==2.0.0 # via stack-data fastjsonschema==2.18.1 # via nbformat -fiddle @ git+https://github.com/google/fiddle +fiddle==0.2.11 # via -r praxis-requirements.in flatbuffers==1.12 # via tensorflow @@ -165,7 +165,7 @@ ipywidgets==8.1.1 # via jupyter isoduration==20.11.0 # via jsonschema -jax @ git+https://github.com/google/jax +jax==0.4.18 # via # -r praxis-requirements.in # chex diff --git a/requirements.in b/requirements.in index 5f8e04fd..e0defcae 100644 --- a/requirements.in +++ b/requirements.in @@ -5,9 +5,9 @@ absl-py clu einops etils -fiddle @ git+https://github.com/google/fiddle +fiddle==0.2.11 flax -jax @ git+https://github.com/google/jax +jax==0.4.18 jax-bitempered-loss jaxtyping lingvo