crowsonkb / mdmm-jaxView external linksLinks
Gradient-based constrained optimization for JAX
☆37Sep 14, 2022Updated 3 years ago
Alternatives and similar repositories for mdmm-jax
Users that are interested in mdmm-jax are comparing it to the libraries listed below
Sorting:
- Repo for the paper: Learning with Muscles: Benefits for Data-Efficiency and Robustness in Anthropomorphic Tasks. https://al.is.mpg.de/pub…☆14Dec 1, 2022Updated 3 years ago
- 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
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆56Sep 18, 2022Updated 3 years ago
- CogView2 for GPUs with 12/16/24GB vRAM☆16Jun 24, 2022Updated 3 years ago
- StyleGAN2 - Official TensorFlow Implementation with practical improvements☆16Sep 11, 2020Updated 5 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Nov 18, 2022Updated 3 years ago
- Orthogonal polynomials with JAX☆27Jan 20, 2026Updated 3 weeks ago
- A JAX library for Density Functional Theory.☆54Nov 25, 2025Updated 2 months ago
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 5 years ago
- Cross-Domain Imitation Learning via Optimal Transport☆25Jun 24, 2022Updated 3 years ago
- Text-writing denoising diffusion (and much more)☆30May 14, 2023Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆29Sep 25, 2021Updated 4 years ago
- Automatic Functional Differentiation in JAX☆81Sep 18, 2025Updated 4 months ago
- Notebooks and other course materials for Emory QTM 340 (Fall 2022)☆12Dec 13, 2022Updated 3 years ago
- Store Terraform state for your GitHub Actions as an encrypted artifact or repository file.☆10Jul 12, 2024Updated last year
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- Official repo of the OpenApePose dataset☆12Feb 6, 2024Updated 2 years ago
- Code repository for "Multi-Task Encoder-Dual-Decoder Modeling Framework on Mixed Frequency Data", International Journal of Forecasting, 2…☆12Feb 18, 2024Updated last year
- ☆35Jul 5, 2023Updated 2 years ago
- A tool for benchmarking image generation models.☆33Jan 13, 2023Updated 3 years ago
- Minimal zero-shot intent classifier for arbitrary intent slot filling, via LLM prompting w LangChain.☆37Mar 13, 2023Updated 2 years ago
- Inferring beliefs about dynamics from behavior☆30May 24, 2018Updated 7 years ago
- ☆78Apr 3, 2023Updated 2 years ago
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆33Sep 15, 2023Updated 2 years ago
- A library for probing Stockfish's NNUEs. The code for reading parameters and forward propagation is taken from Stockfish☆12Nov 18, 2025Updated 2 months ago
- Python library for the enigma machine☆16Mar 7, 2024Updated last year
- workspace comprising demo packages for our roscon2018 talk☆10Dec 21, 2019Updated 6 years ago
- Mathematical foundations of data analysis, Winter semester 22-23☆13Jan 31, 2023Updated 3 years ago
- CEA computes the equilibrium composition of mixtures via free-energy minimization, and uses the resulting product concentrations to deter…☆42Updated this week
- codebase for the SIMAT dataset and evaluation☆38Feb 16, 2022Updated 4 years ago
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆38Jan 25, 2023Updated 3 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆541Feb 9, 2026Updated last week
- Raisimpy: Python wrappers for RaiSim☆39Jun 24, 2020Updated 5 years ago
- Finite element methods for electronic structure calculations on small systems☆42Oct 26, 2025Updated 3 months ago
- A snappy + easy + pretty TUI debugger for Python.☆69Oct 3, 2025Updated 4 months ago
- This is a versatile Python library tailored for the Tehran Stock Exchange, offering robust tools for processing and analyzing TSE symbol …☆13May 16, 2024Updated last year
- AuRaCh - Automated Radiation Chemistry Simulations☆13Mar 7, 2025Updated 11 months ago
- ☆15Oct 24, 2023Updated 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 3 years ago