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