Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆317Apr 21, 2026Updated last 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:
- 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
- PyTorch implementation of Levenberg-Marquardt training algorithm☆97Feb 12, 2026Updated 2 months ago
- A comprehensive list of gradient-based multi-objective optimization algorithms in deep learning.☆108Aug 8, 2025Updated 8 months ago
- Fully customizable Classifer Free Guidance for ComfyUI☆15Jul 14, 2024Updated 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.
- ☆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.☆53Feb 16, 2026Updated 2 months ago
- ☆14Mar 15, 2021Updated 5 years ago
- [NeurIPS'24 Oral] Official repository for the paper "Scale Equivariant Graph Metanetworks"☆23Nov 24, 2024Updated last year
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆19Oct 27, 2021Updated 4 years ago
- 🍓 Build and train energy-based and diffusion models in PyTorch ⚡.☆84Apr 17, 2026Updated last week
- A minimalist implementation of the ViT (Vision Transformer) model, using tinygrad☆16Sep 2, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆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
- 梯度下降与Levenberg-Marquardt算法的比较。Comparison of gradient descent and Levenberg–Marquardt algorithm. Сравнение алгоритма градиентного спуска и…☆31Mar 2, 2020Updated 6 years ago
- A collection of niche / personally useful PyTorch optimizers with modified code.☆28Apr 14, 2026Updated 2 weeks ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆35Jun 17, 2022Updated 3 years ago
- Implementations of the renormalization group-based diffusion model (RGDM).☆16Mar 10, 2025Updated last year
- A very lightweight bevy ui framework with just the most important bits of bevy_sickle_ui☆13Mar 24, 2026Updated last month
- Automated Design of Agentic Systems☆10Sep 7, 2024Updated last year
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆42May 5, 2025Updated 11 months 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.
- ☆14Mar 3, 2022Updated 4 years ago
- Finite element method for solving the electromagnetic Helmholtz equation. Based on scikit-fem and scipy.☆15Mar 12, 2023Updated 3 years ago
- RS-IMLE☆44Dec 7, 2024Updated last year
- Code for the paper "Siamese-GAN: Learning Invariant Representations for Aerial Vehicle Image Categorization"☆13Oct 15, 2018Updated 7 years ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆181Apr 21, 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
- Schedule-Free Optimization in PyTorch☆2,274May 21, 2025Updated 11 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ComfyUI node to apply the ResAdapter Unet patch for SD1.5 models☆31Feb 27, 2025Updated last year
- Stencil with Optimized Dataflow Architecture☆12Feb 27, 2024Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆841Dec 9, 2024Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Apr 8, 2022Updated 4 years ago
- GoLU, a novel, self-gated and element-wise activation function that performs well over a diverse set of tasks☆25Oct 4, 2025Updated 6 months ago
- Code for the paper: Mixed Models with Multiple Instance Learning☆21Nov 10, 2025Updated 5 months ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆148Nov 9, 2023Updated 2 years ago