nitrous-ema. Fast and simple post-hoc EMA (Karras et al., 2023) for PyTorch with minimal `.item()` calls. ~78% lower overhead than ema_pytorch.

github.com/hkchengrex/nitrous-ema

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.