GitHub repository: mctigger/tensorcontainer
Primary language: Python
Description: 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.