7 projects
tensordict-nightly
A dict[str, torch.Tensor] that behaves like one tensor.
torchrl-nightly
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning
torchrl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning
tensordict
TensorDict is a pytorch dedicated tensor container.
llmtrace-lite
Zero-config decorator for tracing LLM calls
trackle-ml
Local-first, git-aware ML experiment tracking.
autoattn
Automatic routing between attention backends (dense, flash, sparse)