Manipulate tensors with PackedSequence and CattedSequence
-
Updated
Jan 4, 2026 - Python
#
ragged-tensor
Here are 2 public repositories matching this topic...
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
Improve this page
Add a description, image, and links to the ragged-tensor topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the ragged-tensor topic, visit your repo's landing page and select "manage topics."