BobMcDear / attorch

A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
MIT License
483 stars 22 forks source link