EelcoHoogendoorn / numga
Geometric algebra in JAX and numpy
☆81Updated last month
Alternatives and similar repositories for numga
Users that are interested in numga are comparing it to the libraries listed below
Sorting:
- Geometric Algebra package for JAX☆49Updated 3 years ago
- A symbolically optimized Geometric Algebra library with PyTorch/NumPy/SymPy/etc. compatibility and ganja.js visualization.☆72Updated last month
- TbGAL: Tensor-Based Geometric Algebra Library☆27Updated 3 years ago
- Stencil computations in JAX☆71Updated last year
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆44Updated 6 months ago
- Turn SymPy expressions into trainable JAX expressions.☆337Updated 3 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆150Updated this week
- Python package for Geometric / Clifford Algebra with TensorFlow☆50Updated last year
- ☆103Updated 10 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 3 weeks ago
- Pytrees + dataclasses ❤️☆62Updated 3 weeks ago
- A functional training loops library for JAX☆88Updated last year
- AutoBound automatically computes upper and lower bounds on functions.☆360Updated 10 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆52Updated 2 weeks ago
- Mathematical operations for JAX pytrees☆199Updated 5 months ago
- Multiple dispatch over abstract array types in JAX.☆121Updated 2 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆126Updated 7 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- JAX Arrays for human consumption☆92Updated last year
- Visualisation library for geometric algebra with cefpython and ganja.js☆52Updated 3 years ago
- Geometric kernels on manifolds, meshes and graphs☆250Updated last month
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆57Updated last year
- Exponential families for JAX☆69Updated last week
- Tools for JAX☆46Updated last week
- Orthogonal polynomials in all shapes and sizes.☆184Updated last year
- The code for "Diffusion Geometry" (2024).☆42Updated 11 months ago
- A Wadler--Lindig pretty printer for Python☆38Updated last week
- Documentation:☆119Updated last year
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 6 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆92Updated this week