IRIS-HEP Fellow: Bo Zheng
Fellowship dates: May - August 2020
Home Institution: Rice University
pyhf Hardware Acceleration Benchmarking with GPUs and TPUsWrite 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 project proposal
- Matthew Feickert (University of Illinois at Urbana-Champaign)
Lukas Heinrich (pyhf dev team)
Giordon Stark (pyhf dev team)