soulitzer 161a2340ee Switch to using Python nested int (#141166) ... Doesn't seem to noticeably slow down eager - TestNestedTensorSubclass tests with and without the PR finished in similar amounts of time (around 57s, 58s) Pull Request resolved: https://github.com/pytorch/pytorch/pull/141166 Approved by: https://github.com/ezyang		2024-12-02 19:17:30 +00:00
..
__init__.py
nested_int.py	Switch to using Python nested int (#141166)	2024-12-02 19:17:30 +00:00
nested_tensor.py	Switch to using Python nested int (#141166)	2024-12-02 19:17:30 +00:00
ops.py	NJT: Return correct number of outputs for chunk() on the batch dim (#141604)	2024-11-27 02:31:23 +00:00
sdpa.py	Misc. non-contig NJT fixes (#140160)	2024-11-09 01:18:26 +00:00