Triton is a programming language and compiler for writing highly-efficient, custom deep-learning primitives developed by

OpenAI OpenAI is an artificial intelligence (AI) research laboratory consisting of the for-profit corporation OpenAI LP and its parent company, the non-profit OpenAI Inc. The company conducts research in the field of AI with the stated goal of promo ...

. The aim of Triton is to provide an open-source environment to write fast code at higher productivity than

CUDA CUDA (or Compute Unified Device Architecture) is a parallel computing platform and application programming interface (API) that allows software to use certain types of graphics processing units (GPUs) for general purpose processing, an approach ca ...

, but also with higher flexibility, than other existing DSLs.

Compatibility

Supported platforms

Linux Linux ( or ) is a family of open-source Unix-like operating systems based on the Linux kernel, an operating system kernel first released on September 17, 1991, by Linus Torvalds. Linux is typically packaged as a Linux distribution, which ...

Supported hardware

Nvidia Nvidia CorporationOfficially written as NVIDIA and stylized in its logo as VIDIA with the lowercase "n" the same height as the uppercase "VIDIA"; formerly stylized as VIDIA with a large italicized lowercase "n" on products from the mid 1990s to ...

GPUs (Compute Capability 7.0+) * Under development: AMD GPUs, CPUs

References

External links

* * Python (programming language) implementations OpenAI {{Prog-lang-stub