A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
Primary LanguagePythonMIT LicenseMIT