Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆392Sep 3, 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…☆123Feb 21, 2025Updated last year
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆27Updated this week
- LibMOON is a standard and flexible framework to study gradient-based multiobjective optimization.☆144Mar 28, 2025Updated last year
- 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☆26Dec 4, 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.
- Implementations of the algorithms described in the paper: On the Convergence Theory for Hessian-Free Bilevel Algorithms.☆11Nov 1, 2024Updated last year
- Efficient optimizers☆340Aug 29, 2026Updated last week
- Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)☆132May 9, 2024Updated 2 years ago
- BANF: Band-limited Neural Fields for Levels of Detail Reconstruction☆20Jul 28, 2025Updated last year
- 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☆427Updated this week
- Official implementation of "Fair Resource Allocation in Multi-Task Learning" [ICML 2024]☆23Oct 22, 2024Updated last year
- [CVPR 2024] Joint-Task Regularization for Partially Labeled Multi-Task Learning☆24May 31, 2024Updated 2 years ago
- torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…☆242Aug 30, 2026Updated last week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- This Logseq plugin is designed to transform LaTex formula images from the clipboard into LaTex code using Transformers.☆14Jul 10, 2024Updated 2 years ago
- ☆17Jun 23, 2023Updated 3 years ago
- Fully customizable Classifer Free Guidance for ComfyUI☆15Jul 14, 2024Updated 2 years ago
- ☆33Jan 6, 2025Updated last year
- ☆12Mar 1, 2022Updated 4 years ago
- Repository for "Training Audio Captioning Models without Audio"☆10Sep 26, 2023Updated 2 years ago
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆51Updated this week
- OpTree: Optimized PyTree Utilities☆213Updated this week
- EDM2 and Autoguidance -- Official PyTorch implementation☆852Dec 9, 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.
- A dataset of 173 progressive metal songs, in both GuitarPro and token formats, as per the specifications in DadaGP.☆18Nov 19, 2024Updated last year
- Learning to optimize (L2O) package that provides basic functionalities to help fit proxy models for optimization.☆16Apr 1, 2025Updated last year
- PyTorch implementation of 'Concrete Dropout'☆21Nov 9, 2023Updated 2 years ago
- maximal update parametrization (µP)☆1,754Jul 17, 2024Updated 2 years ago
- Schedule-Free Optimization in PyTorch☆2,325Jul 28, 2026Updated last month
- My own version of crepe, SOTA pitch tracking tool in PyTorch.☆16Feb 11, 2026Updated 6 months ago
- [NeurIPS'24 Oral] Official repository for the paper "Scale Equivariant Graph Metanetworks"☆23Nov 24, 2024Updated last year
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,864Updated this week
- Supplementary code for the NeurIPS 2020 paper "Matern Gaussian processes on Riemannian manifolds".☆31Feb 3, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Code for "A survey and benchmark of high-dimensional Bayesian optimization of discrete sequences"☆16Feb 24, 2025Updated last year
- TensorDict is a pytorch dedicated tensor container.☆1,037Updated this week
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆21Oct 27, 2021Updated 4 years ago
- Google Hashcode 2021 submissions by our team, notok.☆12Mar 4, 2021Updated 5 years ago
- Learning ReLU INRs with B-spline wavelets.☆14Jun 5, 2024Updated 2 years ago
- Tight Binding Machine Learning Toolkit☆47Dec 2, 2025Updated 9 months ago
- ComfyUI node implementing caching optimizations for Chroma☆19Aug 15, 2025Updated last year