IRIS-HEP Fellow: Bo Zheng
Fellowship dates: May - August 2020
Home Institution: Rice University
Write a hardware acceleration benchmarking suite for the GPU enabled backends (TensorFlow, PyTorch, and JAX) in Python (pyhf Issues 301, 348). Write comparison plot generation code in Python for the performance of the GPU enabled backends against each other and the CPU backends. Thoroughly document the performance benchmarking in the form of a case study. Evaluate the hardware acceleration test suite using Google TPUs. Profile the pyhf codebase to determine where the hardware acceleration is making the largest difference.
More information: My full proposal
Matthew Feickert (IRIS-HEP, pyhf dev team)
Lukas Heinrich (pyhf dev team)
Giordon Stark (pyhf dev team)