Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆390Jul 20, 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:
- 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
- JAX library for neural operator and PDE foundation model training.☆24Updated this week
- This project is a implementation in PyTorch for ZO-AdaMU optimization: Adapting Perturbation with the Momentum and Uncertainty in Zeroth-…☆15Dec 12, 2023Updated 2 years ago
- stochastic bfloat16 based optimizer library☆25Dec 4, 2024Updated last year
- Implementations of the algorithms described in the paper: On the Convergence Theory for Hessian-Free Bilevel Algorithms.☆11Nov 1, 2024Updated last year
- 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.
- Efficient optimizers☆335Jul 11, 2026Updated last week
- Scattering Transform with Random Paths for Machine Learning☆16Apr 9, 2026Updated 3 months ago
- Implementation of paper "Do Wide and Deep Networks Learn the Same Things?"☆16Mar 15, 2022Updated 4 years ago
- optimizer & lr scheduler & loss function collections in PyTorch☆421May 23, 2026Updated last month
- [CVPR 2024] Joint-Task Regularization for Partially Labeled Multi-Task Learning☆24May 31, 2024Updated 2 years ago
- Muon is an optimizer for hidden layers in neural networks☆2,721May 24, 2026Updated last month
- ☆17Jun 23, 2023Updated 3 years ago
- Fast, Modern, and Low Precision PyTorch Optimizers☆129May 16, 2026Updated 2 months ago
- Fully customizable Classifer Free Guidance for ComfyUI☆15Jul 14, 2024Updated 2 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- ☆33Jan 6, 2025Updated last year
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆186May 13, 2026Updated 2 months ago
- 📄Small Batch Size Training for Language Models☆82Mar 18, 2026Updated 4 months ago
- Repository for "Training Audio Captioning Models without Audio"☆10Sep 26, 2023Updated 2 years ago
- OpTree: Optimized PyTree Utilities☆214Updated this week
- EDM2 and Autoguidance -- Official PyTorch implementation☆847Dec 9, 2024Updated last year
- Code for USENIX Security 2023 Paper "Every Vote Counts: Ranking-Based Training of Federated Learning to Resist Poisoning Attacks"☆20May 19, 2024Updated 2 years ago
- A dataset of 173 progressive metal songs, in both GuitarPro and token formats, as per the specifications in DadaGP.☆18Nov 19, 2024Updated last year
- The AdEMAMix Optimizer: Better, Faster, Older.☆188Sep 12, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- A Python Library for Learning Non-Euclidean Representations☆75Jul 9, 2026Updated last week
- PyTorch implementation of 'Concrete Dropout'☆21Nov 9, 2023Updated 2 years ago
- Dockerized cross-compilation for the Bela platform☆14May 24, 2020Updated 6 years ago
- On-going VA modeling research. Modeling dynamic range compressor using S4.☆19Nov 29, 2025Updated 7 months ago
- maximal update parametrization (µP)☆1,740Jul 17, 2024Updated 2 years ago
- Schedule-Free Optimization in PyTorch☆2,315Jun 18, 2026Updated last month
- My own version of crepe, SOTA pitch tracking tool in PyTorch.☆16Feb 11, 2026Updated 5 months ago
- [NeurIPS'24 Oral] Official repository for the paper "Scale Equivariant Graph Metanetworks"☆23Nov 24, 2024Updated last year
- A minimalist implementation of the ViT (Vision Transformer) model, using tinygrad☆17Sep 2, 2024Updated last year
- 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.
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,843Jul 8, 2026Updated last week
- Supplementary code for the NeurIPS 2020 paper "Matern Gaussian processes on Riemannian manifolds".☆31Feb 3, 2025Updated last year
- Helpful tools and examples for working with flex-attention☆1,211Updated this week
- TensorDict is a pytorch dedicated tensor container.☆1,034Updated this week
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆20Oct 27, 2021Updated 4 years ago
- 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 last year
- PolarGrad: A Class of Matrix-Gradient Optimizers from a Unifying Preconditioning Perspective☆18Oct 1, 2025Updated 9 months ago