Skip to content

JAXBench: Add back jaxbench profiling and block size tuning scripts - #52

Open
charleshong3 wants to merge 2 commits into
AI-Hypercomputer:mainfrom
charleshong3:add-jaxbench-device-profiler
Open

JAXBench: Add back jaxbench profiling and block size tuning scripts#52
charleshong3 wants to merge 2 commits into
AI-Hypercomputer:mainfrom
charleshong3:add-jaxbench-device-profiler

Add JAXBench block-size tuner (tune_pallas.py)

e1647de
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jul 10, 2026 in 7s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

e1647de Author: @charleshong3 <cha******ng​@berkeley.edu>

(Only the first commit for a unique contributor is listed.)