Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆328Jun 9, 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
- 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
- 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 11 months ago
- Fully customizable Classifer Free Guidance for ComfyUI☆15Jul 14, 2024Updated 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.
- 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
- [NeurIPS'24 Oral] Official repository for the paper "Scale Equivariant Graph Metanetworks"☆23Nov 24, 2024Updated last year
- ☆11Nov 24, 2025Updated 6 months ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆20Oct 27, 2021Updated 4 years ago
- 🍓 Build and train energy-based and diffusion models in PyTorch ⚡.☆89May 31, 2026Updated last week
- ☆12Mar 1, 2022Updated 4 years ago
- LibMOON is a standard and flexible framework to study gradient-based multiobjective optimization.☆140Mar 28, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Somewhat janky implementation of Sonar Sampling (momentum based sampling) for ComfyUI along with an assortment of advanced noise tools (s…☆49Aug 8, 2025Updated 10 months ago
- 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
- ☆71Sep 5, 2024Updated last year
- Automated Design of Agentic Systems☆10Sep 7, 2024Updated last year
- 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
- Accuracy and fairness trade-offs: A stochastic multi-objective approach☆13Jun 3, 2021Updated 5 years ago
- ☆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
- 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.
- Official code for PEEKABOO2: Adapting Peekaboo with Segment Anything Model for Unsupervised Object Localization in Images and Videos.☆31Dec 27, 2025Updated 5 months ago
- Algorithms for fast alignment of structures in finite and periodic systems.☆15Aug 11, 2022Updated 3 years ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆186May 13, 2026Updated 3 weeks ago
- 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 9 months ago
- Schedule-Free Optimization in PyTorch☆2,299May 18, 2026Updated 3 weeks ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆844Dec 9, 2024Updated last year
- Code for the paper: Mixed Models with Multiple Instance Learning☆22Nov 10, 2025Updated 7 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.
- Massively-Parallel Natural Extension of Reference Frame☆34Jan 18, 2023Updated 3 years ago
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 years ago
- Code repository of the paper "Alleviating Adversarial Attacks on Variational Autoencoders with MCMC" published at NeurIPS 2022. https://a…☆10Dec 14, 2022Updated 3 years ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆152Nov 9, 2023Updated 2 years ago
- Viewer for text datasets in formats like HuggingFace, JSONL, etc.☆15Feb 25, 2025Updated last year
- Efficient optimizers☆330May 13, 2026Updated 3 weeks ago
- Rank binders by structure modeling☆17Apr 10, 2026Updated 2 months ago