Chris Jones

icon company Google icon location Senior Product Manager at Google supporting PyTorch/XLA and OpenXLA. Ex-serial entrepreneur, roboticist, occasional woodworker.

Results 3 comments of Chris Jones

Offline maps

I'm also looking for the ability to export a 3D map as an FBX file.

RFC: Evolving PyTorch/XLA for a more native experience on TPU

We're adopting a more PyTorch native approach and will not be using mark_sharding. Initially we're focused on PyTorch's native distributed APIs (torch.distributed). Once our focus shifts to peak distributed performance...

RFC: Evolving PyTorch/XLA for a more native experience on TPU

Correct, SPMD is not planned for the first release. We're implementing the core collectives (scatter, gather, broadcast, etc) and then using those to implement specific APIs like torch.distributed.fsdp. From a...