pytorch/tensordict
View on GitHub ↗ ★ 1,046 Python
A dict[str, torch.Tensor] that behaves like one tensor, and TensorClass: typed, batched dataclasses that compile.
3 mentions on Hacker News by 2 people: 3 stories, 0 comments · first 2024-08-02 · latest 2024-11-23
20082026
-
▲ 2
TensorDictstory by pizza · 2024-11-23 · 0 comments
-
▲ 2
Show HN: TensorDict, GPU-accelerated Python dictsstory by vmoens · 2024-08-03 · 0 comments
-
▲ 4
TensorDict: A GPU-accelerated Python dictionarystory by vmoens · 2024-08-02 · 2 comments