Tree-Attention-Torch

Tree-Attention-Torch

kyegomez

An implementation of Tree-Attention in PyTorch because it's in JAX for some reason

5 Stars
0 Forks
5 Watchers
Python Language
mit License
68.9 SrcLog Score
Cost to Build
$77.6K
Market Value
$85.4K

Growth over time

2 data points  ·  2026-04-14 → 2026-04-21
Stars Forks Watchers
💬

How do you feel about this project?

Ask AI about Tree-Attention-Torch

Question copied to clipboard

What is the kyegomez/Tree-Attention-Torch GitHub project? Description: "An implementation of Tree-Attention in PyTorch because it's in JAX for some reason". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.

Question is copied to clipboard — paste it after the AI opens.

How to clone Tree-Attention-Torch

Clone via HTTPS

git clone https://github.com/kyegomez/Tree-Attention-Torch.git

Clone via SSH

[email protected]:kyegomez/Tree-Attention-Torch.git

Download ZIP

Download master.zip

Found an issue?

Report bugs or request features on the Tree-Attention-Torch issue tracker:

Open GitHub Issues