G github.com Show HN: Grampax, a "torch.autocast"-style interface for mixed precision in JAX 3 points kazga 1 year ago 0 comments