local-attention-flax

PyPI page
Home page
Author: Phil Wang
License: MIT
Summary: Local Attention - Flax Module in Jax
Latest version: 0.0.2
Required dependencies: einops | flax | jax | jaxlib

Downloads last day: 0
Downloads last week: 8
Downloads last month: 19