Variable-length tensors with Triton kernels at 1.6× of FlashAttention-2 on H100. NumPy / PyTorch / MLX / JAX backends.
python machine-learning deep-learning gpu transformers pytorch triton primitive attention tensor mlx jax flash-attention ragged-tensor varlen
-
Updated
May 24, 2026 - Python