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

Mentors:

• Matthew Feickert (University of Wisconsin–Madison)

• Lukas Heinrich (CERN, pyhf dev team)

• Giordon Stark (UCSC, pyhf dev team)

Presentations and Publications

• 31 Aug 2020 - "pyhf Hardware Acceleration Benchmarking with CPUs and GPUs", Bo Zheng, IRIS-HEP Topical Meetings Recording: pyhf Hardware Acceleration Benchmarking with CPUs and GPUs

---

Current Status
July 2022 - Software Engineer at Google

Contact me:

• Send me an email
  
• Visit my GitHub Repo
  
• Visit my LinkedIn Profile