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.shYou can also install Trident using pip.
pip install git+https://github.com/kakaobrain/trident.gitYou 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