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
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- Python GUI for differential forms☆13Oct 14, 2023Updated 2 years ago
- 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 month
- Variational quantum circuit synthesis☆15Jan 17, 2023Updated 3 years ago
- Efficient numerical calculation of generalized filter functions☆17Mar 1, 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
- Interval Analysis and Mixed Monotone Reachability in JAX☆16Feb 22, 2026Updated 2 weeks ago
- Simple and fast quantum error correction architecture simulator for the rotated surface code☆13Aug 8, 2021Updated 4 years ago
- In this repository we store tools and utilities shared between our repositories.☆13May 27, 2024Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated 2 weeks ago
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 8 months ago
- 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☆16Apr 24, 2022Updated 3 years ago
- reproduces experiments from "Grounding inductive biases in natural images: invariance stems from variations in data"☆17Sep 25, 2024Updated last year
- An easy (and fast) API for popular 3D molecular datasets!☆43Feb 27, 2026Updated last week
- ☆18Aug 24, 2024Updated last year
- Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation☆888Jun 2, 2025Updated 9 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.☆52Jan 23, 2026Updated last month
- Examples calling Warp precompiled (cached) kernels directly from C++ (without Python)☆29Sep 26, 2024Updated last year
- Sum of handwritten digits using Deep Sets, implemented in PyTorch.☆21Apr 17, 2019Updated 6 years ago
- Neural Topic Model via Optimal Transport, ICLR 2021☆16Mar 31, 2021Updated 4 years ago
- Julia Framework for Quantum Dynamics and Control☆58Updated this week
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆34Apr 19, 2025Updated 10 months ago
- Gradient-descent quantum process tomography by learning Kraus operators☆26Aug 4, 2022Updated 3 years ago
- Domain Adaptation benchmark☆27Dec 20, 2024Updated last year
- Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.☆24Feb 11, 2026Updated 3 weeks ago
- ☆34Oct 29, 2025Updated 4 months ago
- Geometry-aware Multilingual Embeddings☆26Dec 8, 2022Updated 3 years ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆710Feb 5, 2026Updated last month
- Jax/Flax rewrite of Karpathy's nanoGPT☆63Feb 15, 2023Updated 3 years ago
- Distributional and Outlier Robust Optimization (ICML 2021)☆28Jul 10, 2021Updated 4 years ago
- A diffusion model for structure-based drug design with faster inference from learned representations of protein structure.☆31Dec 18, 2023Updated 2 years ago
- FBS2e examples in Python☆35Jan 18, 2026Updated last month
- Differentiable Articulated Robotics Library (DARLi)☆27Sep 7, 2024Updated last year
- Julia package for quantum circuit simulation using tensor networks☆40Feb 15, 2022Updated 4 years ago
- Sliding window with Guided Decimation Guessing Decoding for QLDPC codes.☆38May 13, 2025Updated 9 months ago