RobinKa / jaxga
Geometric Algebra package for JAX
☆49Updated 3 years ago
Alternatives and similar repositories for jaxga:
Users that are interested in jaxga are comparing it to the libraries listed below
- Geometric algebra in JAX and numpy☆80Updated 3 weeks ago
- ☆101Updated 9 months ago
- Path Tracing in JAX☆71Updated 5 years ago
- A symbolically optimized Geometric Algebra library with PyTorch/NumPy/SymPy/etc. compatibility and ganja.js visualization.☆67Updated 3 weeks ago
- Python package for Geometric / Clifford Algebra with TensorFlow☆50Updated last year
- Pytrees + dataclasses ❤️☆62Updated 9 months ago
- A functional training loops library for JAX☆86Updated last year
- ☆60Updated 3 years ago
- Exponential families for JAX☆65Updated this week
- TbGAL: Tensor-Based Geometric Algebra Library☆27Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆51Updated last week
- Documentation:☆120Updated last year
- Tools for JAX☆45Updated 2 weeks ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 6 months ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Turning SymPy expressions into JAX functions☆44Updated 4 years ago
- ☆42Updated 3 weeks ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- 🦠 AD in less than 20 lines☆54Updated 3 years ago
- Graph Learning with JAX☆14Updated 2 years ago
- ☆80Updated 3 years ago
- ☆17Updated 7 months ago
- Mathematical operations for JAX pytrees☆199Updated 4 months ago
- ☆246Updated 6 months ago
- You should use PySR to find scaling laws. Here's an example.☆33Updated last year
- Inference Combinators in JAX☆47Updated 5 months ago
- Fast singularity detection with kernel☆33Updated last year
- Stencil computations in JAX☆70Updated last year
- The Energy Transformer block, in JAX☆57Updated last year
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago