← All repos

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