Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆321May 17, 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:
- This project is a implementation in PyTorch for ZO-AdaMU optimization: Adapting Perturbation with the Momentum and Uncertainty in Zeroth-…☆14Dec 12, 2023Updated 2 years ago
- stochastic bfloat16 based optimizer library☆21Dec 4, 2024Updated last year
- 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
- A minimal, single-file implementation of the MeanFlow paper on 2D toy examples, with a side-by-side comparison to rectified flow.☆27Jul 4, 2025Updated 10 months ago
- Exact Pareto Optimal solutions for preference based Multi-Objective Optimization☆68Jun 22, 2022Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Python interface to cd-hit☆10Feb 26, 2019Updated 7 years ago
- A comprehensive list of gradient-based multi-objective optimization algorithms in deep learning.☆110Aug 8, 2025Updated 9 months ago
- Fully customizable Classifer Free Guidance for ComfyUI☆15Jul 14, 2024Updated last year
- ☆17Jun 23, 2023Updated 2 years ago
- Repository for "Training Audio Captioning Models without Audio"☆10Sep 26, 2023Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Aug 21, 2023Updated 2 years ago
- Official implementation of Categorical Flow Maps on text.☆57Feb 16, 2026Updated 3 months ago
- [NeurIPS'24 Oral] Official repository for the paper "Scale Equivariant Graph Metanetworks"☆23Nov 24, 2024Updated last year
- Code for "A survey and benchmark of high-dimensional Bayesian optimization of discrete sequences"☆16Feb 24, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆19Oct 27, 2021Updated 4 years ago
- A minimalist implementation of the ViT (Vision Transformer) model, using tinygrad☆17Sep 2, 2024Updated last year
- ☆12Mar 1, 2022Updated 4 years ago
- Tight Binding Machine Learning Toolkit☆47Dec 2, 2025Updated 5 months ago
- 梯度下降与Levenberg-Marquardt算法的比较。Comparison of gradient descent and Levenberg–Marquardt algorithm. Сравнение алгоритма градиентного спуска и…☆31Mar 2, 2020Updated 6 years ago
- LibMOON is a standard and flexible framework to study gradient-based multiobjective optimization.☆138Mar 28, 2025Updated last year
- ☆16Jun 14, 2024Updated last year
- A collection of niche / personally useful PyTorch optimizers with modified code.☆28Apr 14, 2026Updated last month
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆35Jun 17, 2022Updated 3 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.
- Implementations of the renormalization group-based diffusion model (RGDM).☆16Mar 10, 2025Updated last year
- Automated Design of Agentic Systems☆10Sep 7, 2024Updated last year
- Structure-based fragment identification in latent space☆30Mar 4, 2026Updated 2 months ago
- Goal of this project is to build Classification Decision Trees and Regression Decision trees without using any Machine learning libraries☆10Dec 28, 2018Updated 7 years ago
- Lambda Networks implemented in PyTorch☆13Feb 22, 2021Updated 5 years ago
- Accuracy and fairness trade-offs: A stochastic multi-objective approach☆13Jun 3, 2021Updated 4 years ago
- ComfyUI developer tools (Custom Node)☆24Jan 28, 2026Updated 3 months ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆42May 5, 2025Updated last year
- Finite element method for solving the electromagnetic Helmholtz equation. Based on scikit-fem and scipy.☆15Mar 12, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Algorithms for fast alignment of structures in finite and periodic systems.☆15Aug 11, 2022Updated 3 years ago
- [ICML2025] LoRA fine-tune directly on the INT4 models.☆40Nov 25, 2024Updated last year
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆184May 13, 2026Updated last week
- Source code of "What can linearized neural networks actually say about generalization?☆20Oct 21, 2021Updated 4 years ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Oct 2, 2023Updated 2 years ago
- Official Implementation of Impliti-Zoo: Large-Scale Dataset of Neural Implicit Functions for 2D Images and 3D Scenes☆23Aug 23, 2025Updated 8 months ago
- The repository for Speech Recognition Israel meetup group. It is used to material collection and sharing.☆13Jul 12, 2020Updated 5 years ago