pytorch/docs/source/_static/img
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
..
distributed_autograd
dynamo
fine_grained_apis
inductor_profiling
masked
nested Update OSS nested tensor docs to focus on NJT (#145402) 2025-01-25 04:08:19 +00:00
nn
onnx [ONNX] Update images and APIs to onnx_dynamo.rst (#144358) 2025-01-08 21:44:43 +00:00
pipeline_parallelism
profiling_torch_compile
tensorboard
torch_cuda_memory
aliastracker_graph.png
dynamic_graph.gif
meshgrid.png
pytorch-logo-dark-unstable.png
pytorch-logo-dark.png
pytorch-logo-dark.svg
pytorch-logo-flame.png
pytorch-logo-flame.svg
rpc_arch.png
tensor_illustration.png