Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆309Apr 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:
- 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…☆122Feb 21, 2025Updated last year
- Exact Pareto Optimal solutions for preference based Multi-Objective Optimization☆67Jun 22, 2022Updated 3 years ago
- Python interface to cd-hit☆10Feb 26, 2019Updated 7 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- Fully customizable Classifer Free Guidance for ComfyUI☆15Jul 14, 2024Updated last year
- ☆17Jun 23, 2023Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Aug 21, 2023Updated 2 years ago
- ☆29Jul 1, 2022Updated 3 years ago
- [NeurIPS'24 Oral] Official repository for the paper "Scale Equivariant Graph Metanetworks"☆24Nov 24, 2024Updated last year
- Code for "A survey and benchmark of high-dimensional Bayesian optimization of discrete sequences"☆16Feb 24, 2025Updated last year
- 🍓 Build and train energy-based and diffusion models in PyTorch ⚡.☆83Mar 22, 2026Updated 2 weeks ago
- ☆12Mar 1, 2022Updated 4 years ago
- Somewhat janky implementation of Sonar Sampling (momentum based sampling) for ComfyUI along with an assortment of advanced noise tools (s…☆48Aug 8, 2025Updated 8 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Official code for AAAI 2026 paper (One-Step Generative Policies with Q-Learning: A Reformulation of MeanFlow)☆31Dec 15, 2025Updated 3 months ago
- A collection of niche / personally useful PyTorch optimizers with modified code.☆28Oct 25, 2025Updated 5 months ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆34Jun 17, 2022Updated 3 years ago
- 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 last month
- Accuracy and fairness trade-offs: A stochastic multi-objective approach☆13Jun 3, 2021Updated 4 years ago
- Public code release for the paper "ProCreate, Don’t Reproduce! Propulsive Energy Diffusion for Creative Generation"☆41Nov 30, 2025Updated 4 months ago
- ComfyUI developer tools (Custom Node)☆24Jan 28, 2026Updated 2 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- ☆14Mar 3, 2022Updated 4 years ago
- Open source for SiTunes, a situational music recommendation feedback dataset that includes physiological, psychological, and environmenta…☆10Mar 15, 2024Updated 2 years ago
- ☆21Feb 13, 2026Updated last month
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆173Feb 4, 2026Updated 2 months ago
- RS-IMLE☆44Dec 7, 2024Updated last year
- Official Implementation of Impliti-Zoo: Large-Scale Dataset of Neural Implicit Functions for 2D Images and 3D Scenes☆23Aug 23, 2025Updated 7 months ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Oct 2, 2023Updated 2 years ago
- Schedule-Free Optimization in PyTorch☆2,266May 21, 2025Updated 10 months ago
- ComfyUI node to apply the ResAdapter Unet patch for SD1.5 models☆31Feb 27, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- EDM2 and Autoguidance -- Official PyTorch implementation☆834Dec 9, 2024Updated last year
- Self-supervised learning with graph neural network backbones for seizure detection in iEEG data.☆12Oct 7, 2024Updated last year
- This is the repository for the work "An ensemble mechanism to tackle the heterogeneity in asynchronous federated learning"☆11Nov 19, 2021Updated 4 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Apr 8, 2022Updated 4 years ago
- Partial LaTeX sources of my PhD thesis☆25Mar 11, 2013Updated 13 years ago
- ☆12Mar 5, 2026Updated last month
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 years ago