tmd-layer (pypi): TMD Layer
Keywords: artificial|intelligence|Stochastic|Differential|Equation|attention|mechanism