danielkelshaw/riemax
Riemannian geometry in JAX
This library helps advanced researchers and mathematicians define and work with complex curved spaces, known as manifolds. You input mathematical descriptions of these spaces, and it outputs calculations of geometric properties like curvature or the flow of points along the surface. This is for computational scientists or mathematicians working with advanced geometry in fields like theoretical physics or machine learning.
No commits in the last 6 months.
Use this if you need to perform advanced calculations and define operations on complex, curved geometric spaces within a high-performance numerical computing environment.
Not ideal if you are looking for a general-purpose geometric modeling tool or are new to advanced differential geometry and tensor calculus.
Stars
16
Forks
3
Language
Python
License
MIT
Category
Last pushed
May 14, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/danielkelshaw/riemax"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/