Skip to content

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

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

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