Joel Schlosser b2a0feac85 Update OSS nested tensor docs to focus on NJT (#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: https://github.com/pytorch/pytorch/pull/145402
Approved by: https://github.com/soulitzer

2025-01-25 04:08:19 +00:00

..

_internal

Update NJT linear_backward to return non-aliased tensor bias grad (#145399)

2025-01-25 00:58:04 +00:00

__init__.py

Update OSS nested tensor docs to focus on NJT (#145402)

2025-01-25 04:08:19 +00:00