Tree-Attention-Torch
Python
★ 5
updated 2d ago
An implementation of Tree-Attention in PyTorch because it's in JAX for some reason
No plain-English explanation yet — one is being written right now. Check back in a minute.