A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton. - View it on GitHub