IRIS-HEP Fellow: Bo Zheng



Fellowship dates: May - August 2020
Home Institution: Rice University


pyhf Hardware Acceleration Benchmarking with GPUs and TPUs

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 project proposal

Mentors:
  • Matthew Feickert (University of Illinois at Urbana-Champaign)
  • Lukas Heinrich (pyhf dev team)

  • Giordon Stark (pyhf dev team)



Contact me: