A performance library for machine learning applications.

kakaobrain kakaobrain Last update: Jan 17, 2024

Trident

Welcome

🔱 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

how-to-use

🧑‍💻 Contribution

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

pre-commit install

Subscribe to our newsletter