Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆336Jun 25, 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:
- stochastic bfloat16 based optimizer library☆22Dec 4, 2024Updated last year
- 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
- 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 Lightning implementation of the paper Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and H…☆36Dec 17, 2024Updated last year
- Python interface to cd-hit☆10Feb 26, 2019Updated 7 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.
- ☆17Jun 23, 2023Updated 3 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
- ☆30Jul 1, 2022Updated 3 years ago
- Official implementation of Categorical Flow Maps on text.☆60Feb 16, 2026Updated 4 months ago
- Code for USENIX Security 2023 Paper "Every Vote Counts: Ranking-Based Training of Federated Learning to Resist Poisoning Attacks"☆19May 19, 2024Updated 2 years ago
- ☆15Mar 15, 2021Updated 5 years 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
- 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.
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆20Oct 27, 2021Updated 4 years ago
- A minimalist implementation of the ViT (Vision Transformer) model, using tinygrad☆17Sep 2, 2024Updated last year
- 🍓 Build and train energy-based and diffusion models in PyTorch ⚡.☆96Jun 15, 2026Updated 2 weeks ago
- ☆12Mar 1, 2022Updated 4 years ago
- A collection of niche / personally useful PyTorch optimizers with modified code.☆28Apr 14, 2026Updated 2 months ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆35Jun 17, 2022Updated 4 years ago
- Automated Design of Agentic Systems☆10Sep 7, 2024Updated last year
- Public code release for the paper "ProCreate, Don’t Reproduce! Propulsive Energy Diffusion for Creative Generation"☆41Jun 7, 2026Updated 3 weeks ago
- Official code for AAAI 2026 paper (One-Step Generative Policies with Q-Learning: A Reformulation of MeanFlow)☆39Dec 15, 2025Updated 6 months ago
- 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.
- ComfyUI developer tools (Custom Node)☆24Jan 28, 2026Updated 5 months ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆42May 5, 2025Updated last year
- ☆14Mar 3, 2022Updated 4 years ago
- Official code for PEEKABOO2: Adapting Peekaboo with Segment Anything Model for Unsupervised Object Localization in Images and Videos.☆31Dec 27, 2025Updated 6 months ago
- Byzantine Attack-Resistant Federated Averaging Based on Outlier Elimination☆24Dec 26, 2022Updated 3 years ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆185May 13, 2026Updated last month
- 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.☆33Sep 9, 2025Updated 9 months ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Oct 2, 2023Updated 2 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.
- Official Implementation of Impliti-Zoo: Large-Scale Dataset of Neural Implicit Functions for 2D Images and 3D Scenes☆23Aug 23, 2025Updated 10 months ago
- This repository provides a minimal, single-file implementation of SingLoRA (Single Matrix Low-Rank Adaptation) as described in the paper …☆47Jun 22, 2026Updated last week
- Schedule-Free Optimization in PyTorch☆2,307Jun 18, 2026Updated last week
- 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
- This is the repository for the work "An ensemble mechanism to tackle the heterogeneity in asynchronous federated learning"☆10Nov 19, 2021Updated 4 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Apr 8, 2022Updated 4 years ago