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

Multimodal datasets for spatial intelligence

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.

PyTorch-based Fabric ML framework for training, inference, export, and runtime-aware AI deployment workflows

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

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

Visualize TorchRL's TensorDictModule and TensorDictSequential architectures as dataflow graphs with Graphviz