← all repositories

BobMcDear/attorch

A PyTorch-compatible neural network library written in Python using OpenAI's Triton for GPU kernel optimization.

601 stars Python ML Frameworks
attorch
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

attorch reimplements PyTorch’s nn module using OpenAI’s Triton language for GPU acceleration. It provides optimized implementations of common neural network layers including attention mechanisms, convolutions, and normalization operations. The library supports both training and inference by implementing both forward and backward passes. It targets developers who need custom deep learning operations but lack expertise to write CUDA kernels directly.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.