jax
c3924009 - [pallas] use jax.shard_map in distributed test

Commit

120 days ago

[pallas] use jax.shard_map in distributed test

References

#32535 - [pallas] use jax.shard_map in distributed test

Author

jakevdp

jakevdp

Parents

Loading