tensor-container

Public

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.

compile deep-learning python pytorch tensor tensordict torch

Creat：2025-05-15T19:26:58

Update：2025-07-08T02:23:12

https://github.com/mctigger/tensor-container

Stars

Stars Increase

Related projects

Tensorflow

deep-learning

An Open Source Machine Learning Framework for Everyone

190764

1年前

+14today

AutoGPT

AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.

176969

5个月前

+16today

Stable Diffusion Webui

Stable Diffusion web UI

154559

1年前

+26today

Transformers

bert

? Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

147038

1年前

+47today

30 Seconds Of Code

astro

Coding articles to level up your development skills

124583

3个月前

+21today

Dify

Hot

agent

Dify is an open-source LLM app development platform. Dify's intuitive interface combines AI workflow, RAG pipeline, agent capabilities, model management, observability features and more, letting you quickly go from prototype to production.

107248

3个月前

+151today