5 ms·Show HN: Grampax, a "torch.autocast"-style interface for mixed precision in JAX3 points by kazga 11mo ago