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.

• Platform: Linux
• Hardware: NVIDIA

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

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

pre-commit install