GitHub repository: kyegomez/Tree-Attention-Torch
Primary language: Python
Created: 2024-09-09T19:25:27Z
Description: An implementation of Tree-Attention in PyTorch because it's in JAX for some reason