https://pypi.org/project/jax-linear-attention/ Recent updates to the Python Package Index for jax-linear-attention en https://pypi.org/project/jax-linear-attention/0.1.0/ A JAX LLM training template with Pallas Mosaic GPU kernels for linear attention, state-space models and softmax attention on NVIDIA Hopper Wed, 21 Oct 2026 11:49:20 GMT