EelcoHoogendoorn / numga
Geometric algebra in JAX and numpy
☆70Updated last week
Related projects: ⓘ
- Geometric Algebra package for JAX☆47Updated 2 years ago
- A symbolically optimized and pythonic Clifford (geometric) algebra library named after none other than William Kingdon Clifford.☆41Updated last month
- TbGAL: Tensor-Based Geometric Algebra Library☆26Updated 2 years ago
- ☆135Updated last week
- Mathematical operations for JAX pytrees☆181Updated 4 months ago
- Geometric kernels on manifolds, meshes and graphs☆213Updated last week
- Python package for Geometric / Clifford Algebra with TensorFlow☆48Updated last year
- ☆138Updated 7 months ago
- Multiple dispatch over abstract array types in JAX.☆100Updated last month
- Turn SymPy expressions into trainable JAX expressions.☆313Updated 5 months ago
- A functional training loops library for JAX☆84Updated 7 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆119Updated this week
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆68Updated 3 months ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Stencil computations in JAX☆66Updated 11 months ago
- Code for the book "The Elements of Differentiable Programming".☆57Updated 3 weeks ago
- Differentiable and accelerated spherical transforms with JAX☆123Updated this week
- Exact OU processes with JAX☆30Updated last month
- JAX Arrays for human consumption☆87Updated last year
- Pytrees + dataclasses ❤️☆60Updated 2 months ago
- Use numba-compiled kernels from within Jax☆24Updated this week
- Symbolic Geometric Algebra/Calculus package for SymPy☆227Updated 4 months ago
- AutoBound automatically computes upper and lower bounds on functions.☆355Updated 2 months ago
- ☆24Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆173Updated 2 years ago
- Visualisation library for geometric algebra with cefpython and ganja.js☆50Updated 3 years ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆26Updated last year
- Exponential families for JAX☆54Updated last week
- Run PyTorch in JAX. 🤝☆187Updated 10 months ago
- ☆48Updated last year