Show HN: Grampax, a "torch.autocast"-style interface for mixed precision in JAXgithub.com/davidmarttila3 pointskazga8 months ago