diff --git a/README.md b/README.md index 899d2b2..74f3849 100644 --- a/README.md +++ b/README.md @@ -231,13 +231,13 @@ def avg(rms: torch.Tensor, avg_coef: Union[torch.Tensor, float]): ### Equations -$$ +```math \hat{x}_{\rm rms}[n] = \alpha_{\rm avg} x_{\rm rms}[n] + (1 - \alpha_{\rm avg}) \hat{x}_{\rm rms}[n-1] -$$ +``` ## TODO - [x] CUDA acceleration in Numba - [ ] PyTorch CPP extension - [ ] Native CUDA extension -- [ ] Examples \ No newline at end of file +- [ ] Examples