IRIS-HEP Fellow: Bo Zheng

Fellowship dates: Jun - Aug 2020

Home Institution: Rice University

Project: 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

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

  • Giordon Stark (pyhf dev team)

Contact me: