Show HN: Grampax, a "torch.autocast"-style interface for mixed precision in JAX

3 pointsposted 20 hours ago
by kazga
(github.com)

No comments yet