Reference: https://e3x.readthedocs.io/stable/_autosummary/e3x.nn.functions.gaussian.basic_gaussian.html#e3x.nn.functions.gaussian.basic_gaussian (A Good First Issue for the @lab-cosmo retreat)