acme icon indicating copy to clipboard operation
acme copied to clipboard
verified publisher icon google-deepmind

Fixes #347: Pin JAX/Chex dependencies in Colab notebooks

Open natinew77-creator opened this issue 1 month ago • 0 comments

The quickstart and tutorial notebooks fail to run in Google Colab due to dependency version mismatches with JAX and Chex libraries.

This fix pins compatible versions before installing dm-acme:

  • jax>=0.4.13
  • jaxlib>=0.4.13
  • chex==0.1.7

Closes #347

natinew77-creator avatar Dec 07 '25 16:12 natinew77-creator