nixos-unstable · x86_64-linux · 151fa4e8ddfd · generated 2026-10-07T01:01:54Z

python314Packages.torchtitan 0.3.0 single

PyTorch native platform for training generative AI models

Name
python3.14-torchtitan-0.3.0
Package set
python314Packages
Homepage
https://github.com/pytorch/torchtitan
License
BSD-3-Clause
Source
pkgs/development/python-modules/torchtitan/default.nix:101
Derivation
/nix/store/vfvc1vpc57yqkqr2s8r3naysa37xb8fl-python3.14-torchtitan-0.3.0.drv

Maintainers 1

HandleNameContactVia
GaetanLepage Gaetan Lepage GitHub · gaetan@glepage.com listed on package

Dependencies 37

PackageAsMaintainersUsed by (transitive)Status
python3 3.14.7 propagatedBuildInputs 72,537 team-only
python314Packages.datasets 4.5.0 propagatedBuildInputs 188 single
python314Packages.einops 0.8.2 propagatedBuildInputs 52 single
python314Packages.pillow 12.3.0 propagatedBuildInputs 7,334 single
python314Packages.safetensors 0.8.0 propagatedBuildInputs 133 single
python314Packages.spmd-types 0.2.5 propagatedBuildInputs 1 single
python314Packages.tensorboard 2.21.0 propagatedBuildInputs 19 single
python314Packages.tokenizers 0.23.2 propagatedBuildInputs 181 single
python314Packages.torch 2.13.0 propagatedBuildInputs 397 maintained
python314Packages.torch-checkpointing 0.1.0 propagatedBuildInputs 1 single
python314Packages.torchdata 0.11.0 propagatedBuildInputs 1 single
python314Packages.tyro 1.0.15 propagatedBuildInputs 5 single
python314Packages.wandb 0.30.0 propagatedBuildInputs 57 single
ensureNewerSourcesForZipFilesHook 26.05pre-git nativeBuildInputs
74,596 unmaintained setup hook
python313Packages.pythonOutputDistHook 26.05pre-git nativeBuildInputs
74,564 unmaintained setup hook
python313Packages.pythonRemoveBinBytecodeHook 26.05pre-git nativeBuildInputs
74,584 unmaintained setup hook
python314Packages.expecttest 0.3.0 nativeBuildInputs 44 single
python314Packages.flash-linear-attention 0.5.2 nativeBuildInputs 1 single
python314Packages.pypaBuildHook 26.05pre-git nativeBuildInputs
71,473 unmaintained setup hook
python314Packages.pypaInstallHook 26.05pre-git nativeBuildInputs
71,694 unmaintained setup hook
python314Packages.pytestCheckHook 26.05pre-git nativeBuildInputs
70,075 unmaintained setup hook
python314Packages.pythonCatchConflictsHook 26.05pre-git nativeBuildInputs
71,690 unmaintained setup hook
python314Packages.pythonImportsCheckHook 26.05pre-git nativeBuildInputs
71,715 unmaintained setup hook
python314Packages.pythonMetadataCheckHook 26.05pre-git nativeBuildInputs 71,460 single setup hook
python314Packages.pythonNamespacesHook 26.05pre-git nativeBuildInputs
71,713 unmaintained setup hook
python314Packages.pythonRelaxDepsHook 26.05pre-git nativeBuildInputs
71,476 unmaintained setup hook
python314Packages.pythonRemoveTestsDirHook 26.05pre-git nativeBuildInputs
71,713 unmaintained setup hook
python314Packages.pythonRuntimeDepsCheckHook 26.05pre-git nativeBuildInputs
71,662 unmaintained setup hook
python314Packages.setuptools 83.0.0 nativeBuildInputs 71,381 team-only
python314Packages.tomli-w 1.2.0 nativeBuildInputs 12,460 single
python314Packages.torchvision 0.28.0 nativeBuildInputs 166 single
python314Packages.transformers 5.17.0 nativeBuildInputs 114 maintained
python314Packages.triton 3.7.1 nativeBuildInputs 403 maintained
python314Packages.wrapPython 26.05pre-git nativeBuildInputs
71,719 unmaintained setup hook
writableTmpDirAsHomeHook 26.05pre-git nativeBuildInputs
66,113 unmaintained setup hook
bashNonInteractive 5.3p15 other input 102,441 maintained
stdenv 26.05pre-git other input
90,330 team-only

Used by 0 directly, 0 transitively

No package depends on this one.