Plugin
torchjd
Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
4 skills396
@simplexlabLibrary for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
@simplexlabVerifies that a TorchJD release was published correctly by checking the docs site, installing from PyPI, and smoke-testing newly…
@simplexlabImplements a new method (scalarizer or aggregator) in TorchJD, starting from the research produced by the research-method skill…
@simplexlabPrepares a torchjd package release by verifying the changelog, README, bumping the version in pyproject.toml, and opening a…
@simplexlabResearches about a new method (scalarizer or aggregator) from the scientific literature and how to integrate it to TorchJD. Use…
@simplexlab© 2026 Flowy
Built for Claude Code · Not affiliated with Anthropic