An implementation of Tree-Attention in PyTorch because it's in JAX for some reason
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.
Clone via HTTPS
Clone via SSH
Download ZIP
Download master.zipReport bugs or request features on the Tree-Attention-Torch issue tracker:
Open GitHub Issues