#

tensordict

Here are 4 public repositories matching this topic...

ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

benchmark reinforcement-learning scheduling hydra attention vehicle-routing-problem tsp operations-research cvrp combinatorial-optimization attention-model neural-combinatorial-optimization electronic-design-automation pytorch-lightning torchrl tensordict

Updated Apr 29, 2025
Python

rbyte

yaak-ai / rbyte

Multimodal datasets for spatial intelligence

machine-learning ai robotics pytorch datasets rerun mcap polars tensordict spatial-intelligence

Updated Apr 30, 2025
Python

OriYarden / pytorch_training_optimization_using_tensordict_memory_mapping

Optimizing PyTorch Model Training by Wrapping Memory Mapped Tensors on an Nvidia GPU with TensorDict.

Updated Apr 30, 2025
Python

Kuenoz / pytorch_training_optimization_using_tensordict_memory_mapping

Optimizing PyTorch Model Training by Wrapping Memory Mapped Tensors on an Nvidia GPU with TensorDict.

python ai gpu optimization torch pytorch tensors memory-mapping gpu-optimization pytorch-tensors pytorch-training tensordict memory-mapped-tensors pytorch-training-optimization

Updated May 1, 2025
Python

Improve this page

Add a description, image, and links to the tensordict topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the tensordict topic, visit your repo's landing page and select "manage topics."