triton

Package — Deep LearningPython 3.7+Advanced

OpenAI Triton: write custom GPU kernels in Python

Quick Info

Documentation
Official Docs
Python Version
3.7+
Dependencies
filelock (NVIDIA GPU required)
Install
pip install triton

Learn by Difficulty

Quick Example

python
# Install: pip install triton
import triton

# Basic triton usage
print(f"Using triton")
# See documentation for detailed examples

triton is a third-party package. OpenAI Triton: write custom GPU kernels in Python. Install with: pip install triton

Try in Playground

Tags

packagedeep-learningneural-networkaigpu