SimplexLab/TorchJD

Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).

/ 100

Established

306 stars and 985 monthly downloads. Available on PyPI.

Maintenance 13 / 25

Adoption 17 / 25

Maturity 18 / 25

Community 10 / 25

Stars

306

Forks

Language

Python

License

MIT

Category

Last pushed

Mar 11, 2026

Monthly downloads

985

Commits (30d)

Dependencies

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/SimplexLab/TorchJD"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

Related frameworks

metaopt/torchopt

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.

opthub-org/pytorch-bsf

PyTorch implementation of Bezier simplex fitting

pytorch/xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

clovaai/AdamP

AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)

nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on...