issues
search
pytorch
/
torchrec
Pytorch domain library for recommendation systems
https://pytorch.org/torchrec/
BSD 3-Clause "New" or "Revised" License
1.95k
stars
441
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
[Bug] state_dict returns wrong path when EC is internally DMP wrapped
#2584
JacoCheung
opened
2 hours ago
0
Mark weights unbacked
#2583
Microve
opened
11 hours ago
1
fix broken test
#2582
TroyGarden
closed
17 hours ago
1
add NJT/TD support for EBC and pipeline benchmark
#2581
TroyGarden
opened
21 hours ago
1
Precision session level metric
#2580
iamzainhuda
closed
16 hours ago
2
Output non-pipelined sharded modules from rewrite + enhance pipelinability test
#2579
che-sh
opened
1 day ago
6
Further improvements to EmbeddingBagCollection documentation and related classes
#2578
rodrigodesalvobraz
closed
9 hours ago
3
Simplify scaleup test to remove warn noise
#2577
damianr99
closed
18 hours ago
4
Reduce scaleup build proposal excessive logging
#2576
damianr99
closed
18 hours ago
4
ShardedQuantEmbeddingBagCollection doesn't seem to be distributing the shards properly
#2575
Hanyu-Li
opened
1 day ago
0
minor tweak of metrics
#2574
zhangtemplar
closed
20 hours ago
4
remove unused function _get_unflattened_lengths
#2573
seanx92
closed
1 day ago
1
Remove unused-variable in tiles/test/TilesTestBase.cpp +3
#2572
r-barnes
closed
2 days ago
1
Remove QuantizeEBC grouping by data type
#2571
aporialiao
closed
2 days ago
1
Different nccl stream within EmbeddingKey alltoall and EmbeddingValue…
#2570
zhaozheng09
opened
3 days ago
2
Improve EmbeddingBagCollection documentation
#2569
rodrigodesalvobraz
closed
2 days ago
1
Optimize performance of embeddings sharding
#2568
iamzainhuda
closed
3 days ago
2
Update unittest_ci.yml to linux_job_v2
#2567
atalman
opened
4 days ago
0
Convert directory fbcode/torchrec to use the Ruff Formatter
#2566
tpolasek
closed
2 days ago
4
Dynamic Embeddings CI compatible with linux upgrade
#2565
PaulZhang12
closed
3 days ago
7
Build Linux Wheels skip cuda 12.6
#2564
PaulZhang12
closed
3 days ago
2
[Question/Bug] DP sharding parameters are inconsistent with others.
#2563
JacoCheung
opened
4 days ago
2
Internal ignores for [pytorch][PR] Fix type-safety of `torch.nn.Module` instances
#2562
ezyang
closed
17 hours ago
2
Unify TBE API for key FBGEMM operations (Frontend)
#2561
spcyppt
closed
3 days ago
5
Add `description` to tensor weighted avg metric
#2560
jmschndev
closed
3 days ago
1
Unify InferRwSequenceEmbedding Modules for GPU / CPU
#2559
faran928
opened
1 week ago
6
Remove barriers after PG initialization
#2558
iamzainhuda
closed
5 days ago
3
Support sending using lengths to TBE instead of just offsets
#2557
PaulZhang12
opened
1 week ago
1
Re-shardable Hash Zch (#2538)
#2556
dstaay-fb
closed
13 hours ago
2
Remove deprecated Inference/Quantize Sharded unit test
#2555
aporialiao
closed
1 week ago
1
TorchRec 2D Parallel
#2554
iamzainhuda
closed
21 hours ago
9
Optimize to 1 TBE per rank (no longer separate by quant dtype) for Sharded & Quantized EBC & PEA
#2553
aporialiao
closed
3 days ago
8
Back out "Re-shardable Hash Zch"
#2552
dstaay-fb
opened
1 week ago
1
Favor sharding options with fused_uvm_caching kernel in
#2551
keyan
closed
6 days ago
3
Tighten type hints for tensor arithmetic
#2550
ezyang
closed
1 week ago
1
Remove composable API's fully_shard from torchrec test
#2549
wz337
closed
1 week ago
1
Logging KT's key order warning only once
#2548
gag1jain
opened
1 week ago
3
fix NDCG scatter_reduce_ arg
#2547
iamzainhuda
closed
1 week ago
1
use inverse indices in KJT permute
#2546
iamzainhuda
opened
2 weeks ago
3
Update Dynamic Embedding Wheels CI
#2545
PaulZhang12
opened
2 weeks ago
2
add option for MODULE_ATTR_USE_BATCHING_HINTED_OUTPUT
#2544
seanx92
closed
2 weeks ago
1
permute order cache for qebc
#2543
842974287
opened
2 weeks ago
1
TorchScript bad_alloc issue
#2542
PaulZhang12
opened
2 weeks ago
14
add backward compatible reference for _get_unflattened_lengths
#2541
seanx92
closed
2 weeks ago
1
Support CaliFree and Unweighted NE in TorchRec
#2540
iamzainhuda
closed
2 weeks ago
3
add ddr_cap config to shard_quant_model in TGIF
#2539
zhaojuanmao
closed
2 weeks ago
1
Re-shardable Hash Zch
#2538
dstaay-fb
closed
1 week ago
2
TorchRec inference output dtype
#2537
PaulZhang12
closed
1 week ago
3
make it configurable to use flattened/unflattened length for rebatching
#2536
seanx92
closed
2 weeks ago
4
Split FPEBC
#2535
PaulZhang12
closed
2 weeks ago
2
Next