JAXBench: Benchmarking Autonomous TPU Kernel Optimization
A new benchmark suite called JAXBench, comprising 50 JAX workloads for AI-generated kernel optimization on Google Cloud TPUs, shows that target-specific documentation context matters more than model s…