jax
4a84b913 - lower in_axes=None to XLA replication annotation (#3025)

Commit

5 years ago

lower in_axes=None to XLA replication annotation (#3025) * lower in_axes=None to XLA replication annotation * ignore replicated value for tokens

References

Author

jekbradbury

Parents