https://github.com/google/jax
Raw File
Tip revision: 9e62994bce7c7fcbb2f6a50c9ef89526cd2c2be6 authored by jax authors on 09 May 2024, 19:32:51 UTC
Merge pull request #21135 from hawkinsp:release
Tip revision: 9e62994
nested_pmap.png
image
back to top