JAXBench: curated TPU docs lift AI kernel correctness 5.8% to 37%JAXBench, the first TPU benchmark for AI kernel generat...
JAXBench: curated TPU docs lift AI kernel correctness 5.8% to 37%JAXBench, the first TPU benchmark for AI kernel generation, shows curated docs lift correctness from 5.8% to 37.3% ...