Skip to content

Fix JAX 0.4.31 compatibility, with sharding argument in `convert_el… #406

Fix JAX 0.4.31 compatibility, with sharding argument in `convert_el…

Fix JAX 0.4.31 compatibility, with sharding argument in `convert_el… #406

lint_and_typecheck

succeeded Sep 9, 2024 in 2m 55s