Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆404Nov 5, 2026Updated this week
Alternatives and similar repositories for TorchJD
Users that are interested in TorchJD are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆123Feb 21, 2025Updated last year
- Train models on structured, non-tabular data☆16Updated this week
- building various transformer model architectures and its modules from scratch.☆13Mar 14, 2025Updated last year
- LibMOON is a standard and flexible framework to study gradient-based multiobjective optimization.☆147Mar 28, 2025Updated last year
- Code for "A survey and benchmark of high-dimensional Bayesian optimization of discrete sequences"☆18Feb 24, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- This project is a implementation in PyTorch for ZO-AdaMU optimization: Adapting Perturbation with the Momentum and Uncertainty in Zeroth-…☆15Dec 12, 2023Updated 2 years ago
- A comprehensive list of gradient-based multi-objective optimization algorithms in deep learning.☆120Aug 8, 2025Updated last year
- Efficient optimizers☆340Sep 9, 2026Updated last month
- Implementations of the algorithms described in the paper: On the Convergence Theory for Hessian-Free Bilevel Algorithms.☆11Nov 1, 2024Updated 2 years ago
- Fully customizable Classifer Free Guidance for ComfyUI☆15Jul 14, 2024Updated 2 years ago
- Official Pytorch implementation of the fundamental frequency estimator described in "Robust and Lightweight F0 Estimation Through Mid-Lev…☆14Apr 30, 2026Updated 6 months ago
- A general training alignment framework for LLM training debugging. OSDI ’26 Best Paper Nominee.☆21Oct 20, 2026Updated 2 weeks ago
- A PyTorch Library for Multi-Task Learning☆2,598May 14, 2025Updated last year
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆14Apr 20, 2026Updated 6 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Schedule-Free Optimization in PyTorch☆2,333Jul 28, 2026Updated 3 months ago
- Scattering Transform with Random Paths for Machine Learning☆16Updated this week
- Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)☆135May 9, 2024Updated 2 years ago
- Implementation of paper "Do Wide and Deep Networks Learn the Same Things?"☆16Mar 15, 2022Updated 4 years ago
- A pip installable package for optimal transport inspired loss functions in the spectral domain. Can be used for audio applications such a…☆32Jun 18, 2026Updated 4 months ago
- BANF: Band-limited Neural Fields for Levels of Detail Reconstruction☆20Jul 28, 2025Updated last year
- The AdEMAMix Optimizer: Better, Faster, Older.☆187Sep 12, 2024Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Aug 21, 2023Updated 3 years ago
- optimizer & lr scheduler & loss function collections in PyTorch☆434Oct 30, 2026Updated last week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Code for the paper "Rigid-Body Sound Synthesis with Differentiable Modal Resonators"☆15May 24, 2023Updated 3 years ago
- Muon is an optimizer for hidden layers in neural networks☆2,928May 24, 2026Updated 5 months ago
- Annealed Multiple Choice Learning: Overcoming limitations of Winner-takes-all with annealing (NeurIPS 2024)☆10May 26, 2025Updated last year
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆19Mar 15, 2024Updated 2 years ago
- torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…☆251Oct 22, 2026Updated 2 weeks ago
- Exact Pareto Optimal solutions for preference based Multi-Objective Optimization☆69Jun 22, 2022Updated 4 years ago
- Paret-Front Diverse Batch Multi-Objective Bayesian Optimization☆11Mar 31, 2024Updated 2 years ago
- Official implementation of Scalable Bilevel Loss Balancing for Multi-Task Learning.☆14May 15, 2025Updated last year
- A pure and fast NumPy implementation of Mamba with cache support.☆18Jun 16, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆12Mar 1, 2022Updated 4 years ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆189May 13, 2026Updated 5 months ago
- ☆11Jul 21, 2025Updated last year
- Repository for "Training Audio Captioning Models without Audio"☆10Sep 26, 2023Updated 3 years ago
- A dict[str, torch.Tensor] that behaves like one tensor, and TensorClass: typed, batched dataclasses that compile.☆1,042Updated this week
- This is the official implementation of our multi-channel multi-speaker multi-spatial neural audio codec architecture.☆55Mar 17, 2025Updated last year
- Leveraging Protein Language Models and Structure-Based protein design with Genetic Algorithms☆13Feb 20, 2026Updated 8 months ago