Skip to content

Commit

Permalink
change jax python version
Browse files Browse the repository at this point in the history
  • Loading branch information
@priyakasimbeg
priyakasimbeg committed Jan 16, 2025
1 parent 01eb881 commit c9b6411
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions setup.cfg
View file
Original file line number Diff line number Diff line change
Expand Up @@ -121,8 +121,8 @@ jax_core_deps =

# JAX CPU
jax_cpu =
jax==0.4.38
jaxlib==0.4.38
jax==0.4.36
jaxlib==0.4.36
%(jax_core_deps)s

# JAX GPU
Expand Down

0 comments on commit c9b6411

Please sign in to comment.