Skip to content

Commit b2a0fea

Browse files
jbschlosserpytorchmergebot
authored andcommitted
Update OSS nested tensor docs to focus on NJT (pytorch#145402)
Updated nested tensor docs to be NJT-centric (instead of NST-centric). They now include: * High-level description of NST vs. NJT + a recommendation to use NJT * General NJT construction / usage * torch.compile() integration w/ dynamic shapes * Common errors and how to fix them * Contribution guide * Data layout / shape information (with diagram) * Links to more extensive tutorials involving Transformers / SDPA / FlexAttention Pull Request resolved: pytorch#145402 Approved by: https://github.com/soulitzer
1 parent 392dc17 commit b2a0fea

File tree

3 files changed

+454
-189
lines changed

3 files changed

+454
-189
lines changed

Diff for: docs/source/_static/img/nested/njt_visual.png

53.8 KB
Loading

0 commit comments

Comments
 (0)