Skip to content
@TorchJD

TorchJD

Pinned Loading

  1. torchjd torchjd Public

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

    Python 234 6

Repositories

Showing 4 of 4 repositories
  • torchjd Public

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

    TorchJD/torchjd’s past year of commit activity
    Python 234 MIT 6 1 2 Updated May 14, 2025
  • arena Public

    This repo is used to compare different torchjd functions (e.g. aggregators), potentially from different commits of torchjd, on a variety of topics (computation time, precision, etc.)

    TorchJD/arena’s past year of commit activity
    Python 1 MIT 0 0 0 Updated May 14, 2025
  • documentation Public

    Repository for hosting the built documentation of torchjd

    TorchJD/documentation’s past year of commit activity
    HTML 1 MIT 0 0 0 Updated May 13, 2025
  • trajectories Public

    Optimization of simple functions with JD and plot of the trajectories.

    TorchJD/trajectories’s past year of commit activity
    Python 2 0 0 0 Updated May 4, 2025

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Top languages

Loading…

Most used topics

Loading…