Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆306Mar 14, 2026Updated last week
Alternatives and similar repositories for TorchJD
Users that are interested in TorchJD are comparing it to the libraries listed below
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
- A comprehensive list of gradient-based multi-objective optimization algorithms in deep learning.☆101Aug 8, 2025Updated 7 months ago
- ☆18Jun 23, 2023Updated 2 years ago
- Official implementation of Categorical Flow Maps on text.☆47Feb 16, 2026Updated last month
- 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
- Official code for AAAI 2026 paper (One-Step Generative Policies with Q-Learning: A Reformulation of MeanFlow)☆26Dec 15, 2025Updated 3 months ago
- [NeurIPS'24 Oral] Official repository for the paper "Scale Equivariant Graph Metanetworks"☆24Nov 24, 2024Updated last year
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆19Oct 27, 2021Updated 4 years 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 7 months ago
- LibMOON is a standard and flexible framework to study gradient-based multiobjective optimization.☆128Mar 28, 2025Updated 11 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
- ☆72Sep 5, 2024Updated last year
- Automated Design of Agentic Systems☆10Sep 7, 2024Updated last year
- Structure-based fragment identification in latent space☆29Mar 4, 2026Updated 2 weeks ago
- ☆20Feb 13, 2026Updated last month
- Official code for PEEKABOO2: Adapting Peekaboo with Segment Anything Model for Unsupervised Object Localization in Images and Videos.☆30Dec 27, 2025Updated 2 months ago
- Open source for SiTunes, a situational music recommendation feedback dataset that includes physiological, psychological, and environmenta…☆10Mar 15, 2024Updated 2 years ago
- RS-IMLE☆44Dec 7, 2024Updated last year
- Source code of "What can linearized neural networks actually say about generalization?☆20Oct 21, 2021Updated 4 years ago
- Implementation of the Hierarchical Reasoning Model (HRM), applied to a pathfinding task, plus performance study.☆32Sep 9, 2025Updated 6 months ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Oct 2, 2023Updated 2 years ago
- Schedule-Free Optimization in PyTorch☆2,265May 21, 2025Updated 10 months ago
- ComfyUI node to apply the ResAdapter Unet patch for SD1.5 models☆31Feb 27, 2025Updated last year
- Learning stochastic dynamics from snapshots through regularized unbalanced optimal transport (ICLR'25 oral)☆32Dec 10, 2025Updated 3 months ago
- Self-supervised learning with graph neural network backbones for seizure detection in iEEG data.☆11Oct 7, 2024Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Apr 8, 2022Updated 3 years ago
- ☆12May 26, 2019Updated 6 years ago
- GoLU, a novel, self-gated and element-wise activation function that performs well over a diverse set of tasks☆24Oct 4, 2025Updated 5 months ago
- Efficient optimizers☆294Updated this week
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆146Nov 9, 2023Updated 2 years ago
- Viewer for text datasets in formats like HuggingFace, JSONL, etc.☆15Feb 25, 2025Updated last year
- ☆12Mar 5, 2026Updated 2 weeks ago
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 3 years ago
- Code for ijcai-24 paper "Federated Adaptation for Foundation Model-based Recommendations"☆12Apr 18, 2025Updated 11 months ago
- Rethinking Disparity: A Depth Range Free Multi-View Stereo Based on Disparity☆21Mar 11, 2024Updated 2 years ago