Fix JAX 0.4.31 compatibility, with sharding
argument in `convert_el…
#405
Job | Run time |
---|---|
44s | |
44s | |
2m 26s | |
16s | |
4m 10s |
sharding
argument in `convert_el…
#405
Job | Run time |
---|---|
44s | |
44s | |
2m 26s | |
16s | |
4m 10s |