tensorcontainer. A collection of efficient, type-safe, and torch.compile supported tensor containers for PyTorch. Designed for seamless PyTree integration, offering TensorDict and TensorDataClass for robust and flexible tensor management.

github.com/mctigger/tensorcontainer

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.