SaitejaUtpala / rieoptaxView external linksLinks
Riemannian Optimization Using JAX
☆54Oct 30, 2023Updated 2 years ago
Alternatives and similar repositories for rieoptax
Users that are interested in rieoptax are comparing it to the libraries listed below
Sorting:
- GRAPE with autograd on the TDSE and LME in Python☆12Mar 26, 2024Updated last year
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated last year
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Design and optimize variational quantum sensors☆11Dec 10, 2024Updated last year
- This repo contains papers, books, tutorials and resources on Riemannian optimization.☆56Feb 4, 2026Updated last week
- A Python library to integrate automatic differentiation tools such as JAX with QuTiP and related quantum software packages.☆44May 8, 2022Updated 3 years ago
- ☆15Jan 18, 2021Updated 5 years ago
- Pauli operators manipulation toolbox.☆17Updated this week
- Julia package for contraction of tensor networks, based on the sweep line algorithm outlined in the paper "General tensor network decodin…☆55Oct 18, 2023Updated 2 years ago
- Repository Associated with the Paper☆15Apr 24, 2022Updated 3 years ago
- Differential Programming and Natural Evolutationary Strategies optimization algorithms for transport Majorana Zero Modes.☆14Nov 27, 2020Updated 5 years ago
- reproduces experiments from "Grounding inductive biases in natural images: invariance stems from variations in data"☆17Sep 25, 2024Updated last year
- Special functions in Jax☆16Mar 1, 2025Updated 11 months ago
- ☆18Aug 24, 2024Updated last year
- Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation☆887Jun 2, 2025Updated 8 months ago
- A library for programmatically generating equivariant layers through constraint solving☆281May 8, 2023Updated 2 years ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆50Jan 23, 2026Updated 3 weeks ago
- Code for "Learning Unitary Operators with Help From u(n)", AAAI-17. (https://arxiv.org/abs/1607.04903)☆17Jan 10, 2017Updated 9 years ago
- Neural Topic Model via Optimal Transport, ICLR 2021☆16Mar 31, 2021Updated 4 years ago
- Examples calling Warp precompiled (cached) kernels directly from C++ (without Python)☆28Sep 26, 2024Updated last year
- Franka Panda Bi-Manual Manipulation simulated on MuJoCo☆30Aug 8, 2024Updated last year
- Differentiable simulation of superconducting quantum processors.☆33Dec 6, 2025Updated 2 months ago
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆32Apr 19, 2025Updated 9 months ago
- Gradient-descent quantum process tomography by learning Kraus operators☆26Aug 4, 2022Updated 3 years ago
- ☆34Oct 29, 2025Updated 3 months ago
- Quantum circuit optimization with deep learning☆26May 25, 2020Updated 5 years ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆700Feb 5, 2026Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆115Feb 3, 2026Updated last week
- Jax/Flax rewrite of Karpathy's nanoGPT☆63Feb 15, 2023Updated 2 years ago
- Sliding window with Guided Decimation Guessing Decoding for QLDPC codes.☆38May 13, 2025Updated 9 months ago
- Julia package for quantum circuit simulation using tensor networks☆40Feb 15, 2022Updated 3 years ago
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆25Aug 16, 2023Updated 2 years ago
- Toolset for control, calibration and characterization of physical systems☆71Mar 5, 2024Updated last year
- Biological Relationships - Biorels data preparation infrastructure for biology and drug discovery☆14May 19, 2025Updated 8 months ago
- Riemannian Convex Potential Maps☆67Mar 31, 2023Updated 2 years ago
- ☆30Jan 23, 2023Updated 3 years ago
- OpTaS: An optimization-based task specification library for trajectory optimization and model predictive control.☆138Nov 1, 2025Updated 3 months ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Oct 19, 2022Updated 3 years ago