pytorch/tensordict

A dict[str, torch.Tensor] that behaves like one tensor, and TensorClass: typed, batched dataclasses that compile.

4 best points · 2 comments · 3 Hacker News posts · 1,042 GitHub stars · View on GitHub

Hacker News discussions

• TensorDict: A GPU-accelerated Python dictionary (4 points, 2 comments, by vmoens, 2024-08-02)
• Show HN: TensorDict, GPU-accelerated Python dicts (2 points, 0 comments, by vmoens, 2024-08-03)
• TensorDict (2 points, 0 comments, by pizza, 2024-11-23)