About the requirement of jax version. #84

Activity

1. claudiofantacci commented on Oct 7, 2023

   

   claudiofantacci

   on Oct 7, 2023

   Collaborator

   More actions

   Hi @AlbertHuyb, I think we expect a version >= 0.2.17 (although this might be outdated, I need to properly check it) and I think that if you pip install dm-pix you will get Jax transitively from chex, see here. This is not ideal, I understand it, but I don't think there's anything that prevent you from using the particular version that you require. You can always give it a go and see what happens 😄 Also, you probably gain much more if you install jax yourself with the optimization that suits best your need as described in here (and that's also why we don't enforce installation through pip, but we get it transitively from chex). Hope this helps! 🙏
2. claudiofantacci

   added

   questionFurther information is requestedFurther information is requested

   on Oct 7, 2023
3. google-deepmind

   locked and limited conversation to collaborators on Sep 30, 2024
4. claudiofantacci

   converted this issue into a discussion #106 on Sep 30, 2024