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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
lint_and_typecheck	2m 55s
unit_tests	45s
	3m 40s