Gradient-based constrained optimization for JAX
☆37Sep 14, 2022Updated 4 years ago
Alternatives and similar repositories for mdmm-jax
Users that are interested in mdmm-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Repo for the paper: Learning with Muscles: Benefits for Data-Efficiency and Robustness in Anthropomorphic Tasks. https://al.is.mpg.de/pub…☆15Dec 1, 2022Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆62Sep 26, 2026Updated last month
- A jax-based reimplementation of MBPO☆10May 8, 2025Updated last year
- AI-native open-source finite-element platform connecting engineering, computation, data, and AI.☆33Updated this week
- Heap Implementation for Python☆10Nov 4, 2016Updated 9 years ago
- 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.
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆57Sep 18, 2022Updated 4 years ago
- This repository shows how to efficiently process variable-length sequences in TensorFlow.☆14Apr 26, 2022Updated 4 years ago
- ☆16Jun 3, 2025Updated last year
- utilities to facilitate working with codebases that don't ascribe to normal package management paradigms, e.g. ML research code that can …☆13Nov 26, 2022Updated 3 years ago
- Robust Reinforcement Learning Suite☆36Dec 24, 2024Updated last year
- ☆14Apr 15, 2025Updated last year
- Code for the submission of the team AL4Muscles to the MyoChallenge 2022☆16Mar 12, 2023Updated 3 years ago
- CogView2 for GPUs with 12/16/24GB vRAM☆16Jun 24, 2022Updated 4 years ago
- Code for paper: End-to-end Stochastic Optimization with Energy-based Model☆16Feb 14, 2023Updated 3 years ago
- 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.
- ☆13Jul 30, 2022Updated 4 years ago
- JAX models for ML☆12Mar 2, 2026Updated 8 months ago
- StyleGAN2 - Official TensorFlow Implementation with practical improvements☆16Sep 11, 2020Updated 6 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆48Sep 18, 2026Updated last month
- Automatic Sparse Differentiation in JAX.☆67Oct 21, 2026Updated last week
- Robust and safe deep reinforcement learning algorithms☆17Mar 27, 2024Updated 2 years ago
- This repo contains the code for the paper "Data-driven discovery of multiscale chemical reactions governed by the law of mass action"☆10Mar 2, 2022Updated 4 years ago
- Pytorch Implementation of Deep Kalman Filter☆12Sep 30, 2025Updated last year
- Computational model reported in: Izquierdo EJ, Beer RD. 2018 From head to tail: a neuromechanical model of forward locomotion in Caenorha…☆15Apr 24, 2021Updated 5 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- JAXMg: A multi-GPU linear solver in JAX☆113Oct 22, 2026Updated last week
- This repository provides a Matlab interface to Woltring's classic generalized, cross-validatory (GCV) spline smoothing and differentiatio…☆10Jan 25, 2023Updated 3 years ago
- Cross-Domain Imitation Learning via Optimal Transport☆27Jun 24, 2022Updated 4 years ago
- A generic flux/parameter population synthesis code☆14Sep 16, 2026Updated last month
- A JAX-based Differentiable Density Functional Theory Framework for Materials☆53Apr 20, 2026Updated 6 months ago
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆33Updated this week
- Fast QP Solver in JAX☆28Aug 29, 2024Updated 2 years ago
- ☆246Oct 23, 2026Updated last week
- A library for high-level algorithmic differentiation☆18Apr 23, 2026Updated 6 months 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.
- Library for construction, manipulation and evaluation of factorable functions☆19Updated this week
- ☆11Updated this week
- A JAX library for Density Functional Theory.☆57Nov 25, 2025Updated 11 months ago
- ☆35Apr 8, 2023Updated 3 years ago
- ☆16Oct 31, 2025Updated last year
- ☆10Sep 19, 2023Updated 3 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆631Aug 11, 2026Updated 2 months ago