diff --git a/environment.yml b/environment.yml index 22523a16..c08e1880 100644 --- a/environment.yml +++ b/environment.yml @@ -18,4 +18,4 @@ dependencies: # Pinned: jax 0.11.1 hangs lax.fori_loop on the CPU path (QuantEcon/workspace-lectures#49), # and two lectures (ak_aiyagari, two_computation) skip their own `!pip install jax` cell # because the English runner preinstalls JAX; this CI has to as well. - - jax==0.11.0 + - jax==0.11.2