Show HN: Grampax, a "torch.autocast"-style interface for mixed precision in JAXgithub.com·3 pts·kazga·0