Notice:This resource is provided by a third-party author. Please review the code with AI tools or manually before use to ensure security and compatibility.
PythonBobMcDear/attorch
attorch
A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.