Skip to content

Pull requests: pytorch/torchrec

New pull request New
332 Open 2,405 Closed
332 Open 2,405 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add missing fields to KJT's PyTree flatten/unflatten logic for VBE KJT CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2952 opened May 7, 2025 by jd7-tr Loading…
1
enable cudagraph for AUCMetricComputation CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2951 opened May 7, 2025 by BoyuanFeng Loading…
1
Create a new tensor arg for stride_per_key_per_rank to facilitate torch.export CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2950 opened May 6, 2025 by jd7-tr Loading…
2
Update KJT stride calculation logic to be based off of inverse_indices for VBE KJTs. CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2949 opened May 6, 2025 by jd7-tr Loading…
2
Add missing Pyre mode headers] [shard:2/N] CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2947 opened May 6, 2025 by facebook-github-bot Loading…
1
DistributedModelParallel resharding Interface CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2945 opened May 6, 2025 by aporialiao Loading…
4
Add padding in dynamic sharding for tensors before all2all CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2944 opened May 5, 2025 by aporialiao Loading…
3
Add support for pipelined postproc CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2940 opened May 3, 2025 by che-sh Loading…
3
{Tests-WIP}[Torchrec] Add context manager to use next batch context for postprocs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2939 opened May 3, 2025 by che-sh Loading…
1
Add raw embedding streaming needed params in trec and mvai CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2935 opened May 2, 2025 by chouxi Loading…
3
Extend FSDP1 global clipping support for optimizers other than Shampoo CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2931 opened May 1, 2025 by wz337 Loading…
1
Add logic to stream weights in EmbeddingKVDB CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2930 opened Apr 30, 2025 by chouxi Loading…
1
Add enable_raw_embedding_streaming from TBE config to EmbeddingKVDB CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2928 opened Apr 30, 2025 by chouxi Loading…
5
Allow a context manager to be called around apply_jit CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2927 opened Apr 30, 2025 by kddnewton Loading…
2
Update the _maybe_compute_stride_kjt logic to calculate stride based off of inverse_indices for VBE KJTs. CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2925 opened Apr 29, 2025 by jd7-tr Loading…
3
Move util function for creating VBE KJT to the VbModelInput class CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2906 opened Apr 22, 2025 by jd7-tr Loading…
1
Test include stride_per_key to KJT's flatten and unflatten CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2903 opened Apr 22, 2025 by jd7-tr Loading…
7
Add unsharded module reference to sharded modules CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2901 opened Apr 22, 2025 by aporialiao Loading…
5
Enable custom perf models in MemoryBalancedPartitioner CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2897 opened Apr 19, 2025 by micrain Loading…
1
Perf Model Modeling Critical Path of Sparse Arch CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2896 opened Apr 19, 2025 by micrain Loading…
2
Back out "Fuse "states" level tensors to reduce all gather during metrics compute" CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2893 opened Apr 17, 2025 by iamzainhuda Loading…
1
Bucket offsets and sizes in torchrec shard metadata for bucket wise sharding (#2884) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2885 opened Apr 13, 2025 by faran928 Loading…
9
support bucket offsets and sizes in shard metadata CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2884 opened Apr 12, 2025 by faran928 Loading…
3
create rfc for flexible collision free embedding table proposal CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2883 opened Apr 11, 2025 by emlin Loading…
1
Resolve Pyre CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2880 opened Apr 11, 2025 by aporialiao Loading…
2
ProTip! Filter pull requests by the default branch with base:main.