Notes:
- NAT depends on NATTEN's implementation of Neighborhood Attention.