GitHub repository: sugatoray/tensordict
Description: TensorDict is a pytorch dedicated tensor container.