Show HN: Grampax, a "torch.autocast"-style interface for mixed precision in JAX github.com 3 points by kazga a day ago