kakaobrain / trident

A performance library for machine learning applications.

🔱 Overview

Trident is a performance library for machine learning applications intended for acceleration of training and inference. Trident includes highly optimized kernels, functions and modules for machine learning. Trident is implemented based on OpenAI Triton. Check Wiki for more detail.

💻 Requirements

• Platform: Linux
• Hardware: NVIDIA

📦 Installation

bash install_package.sh

You can also install Trident using pip.

pip install git+https://github.com/kakaobrain/trident.git

You can add Trident in requirements.txt.

trident@git+https://github.com/kakaobrain/trident.git@main

📖 How to use

🧑‍💻 Contribution

You must set up pre-commit to maintain the consistent coding style.

pre-commit install