https://github.com/kakaobrain/trident

A performance library for machine learning applications.
https://github.com/kakaobrain/trident

ai deep-learning library machine-learning performance python pytorch triton

Last synced: 9 months ago
JSON representation

A performance library for machine learning applications.

• Host: GitHub
• URL: https://github.com/kakaobrain/trident
• Owner: kakaobrain
• License: apache-2.0
• Archived: true
• Created: 2023-04-30T01:01:46.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-10-12T10:31:40.000Z (over 2 years ago)
• Last Synced: 2025-03-13T19:22:43.506Z (11 months ago)
• Topics: ai, deep-learning, library, machine-learning, performance, python, pytorch, triton
• Language: Python
• Homepage: https://www.kakaobrain.com
• Size: 514 KB
• Stars: 183
• Watchers: 3
• Forks: 12
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE
  • Codeowners: .github/CODEOWNERS

Awesome Lists containing this project

README

          
![Trident](https://github.com/kakaobrain/trident/assets/7459074/aee6bb37-45a4-4b8e-bf14-0ea63dc574a0)

![Welcome](https://capsule-render.vercel.app/api?type=rect&color=001eff&text=👋%20Hello%20and%20Welcome&fontColor=ffffff)

## 🔱 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](https://github.com/openai/triton). Check [Wiki](https://github.com/kakaobrain/trident/wiki) for more

detail.

## 💻 Requirements

- Platform: Linux

- Hardware: NVIDIA

## 📦 Installation

```bash

bash install_package.sh

```

You can also install Trident using pip.

```bash

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

```

You can add Trident in `requirements.txt`.

```text

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

```

## 📖 How to use

![how-to-use](https://github.com/kakaobrain/trident/assets/7459074/2876a2ca-d7cc-4d60-be81-b8d58cbb9f17)

## 🧑‍💻 Contribution

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

```bash

pre-commit install

```