Enabling PyTorch on XLA Devices (e.g. Google TPU)
Role in this project:
Back-end Developer Contributions:924 reviews, 240 commits, 550 PRs in 11 months
Contributions summary:Will's contributions primarily focused on enhancing the `pytorch/xla` repository, which aims to enable PyTorch on XLA devices. Their work involved implementing and optimizing `torch.linspace()` functionality within the XLA environment. The user modified files related to XLA tensor operations, including changes to the aten_xla_type.cpp and linspace.cpp files, and added corresponding tests. These modifications involved integrating the new implementation and ensuring compatibility with XLA devices.
pytorchxladeep-learningtpucompiler
Models and examples built with TensorFlow
Role in this project:
ML Engineer Contributions:16 commits, 2 pushes, 8 branches in 1 year
Contributions summary:Will primarily contributed to the MNIST model within the TensorFlow models repository. Their work included adding and updating the 2.x version of the MNIST model, and fixing the test file. The user also made internal changes, including moving the mnist folder and modifying the resnet_ctl_imagenet_main.py file. These changes suggest a focus on model development and maintenance within the TensorFlow ecosystem.
deep-learningtensorflow