davidmarttila/grampax

Granular Automatic Mixed Precision for JAX

3 best points · 0 comments · 1 Hacker News post · 6 GitHub stars · View on GitHub

Hacker News discussions

• Show HN: Grampax, a "torch.autocast"-style interface for mixed precision in JAX (3 points, 0 comments, by kazga, 2025-10-31)