RobinKa / jaxgaLinks
Geometric Algebra package for JAX
☆50Updated 3 years ago
Alternatives and similar repositories for jaxga
Users that are interested in jaxga are comparing it to the libraries listed below
Sorting:
- Geometric algebra in JAX and numpy☆83Updated 4 months ago
- Python package for Geometric / Clifford Algebra with TensorFlow☆52Updated 2 years ago
- ☆247Updated last month
- Documentation:☆121Updated 2 years ago
- Pytrees + dataclasses ❤️☆67Updated 3 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- ☆105Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆56Updated 2 weeks ago
- Path Tracing in JAX☆71Updated 5 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- A functional training loops library for JAX☆88Updated last year
- 🦠 AD in less than 20 lines☆54Updated 4 years ago
- A symbolically optimized Geometric Algebra library with PyTorch/NumPy/SymPy/etc. compatibility and ganja.js visualization.☆90Updated this week
- AutoBound automatically computes upper and lower bounds on functions.☆361Updated last year
- Mathematical operations for JAX pytrees☆199Updated 8 months ago
- ☆43Updated 2 months ago
- ☆20Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 10 months ago
- [Archived] A scalable differentiable probabilistic Datalog engine, with Rust☆73Updated 3 years ago
- Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep R…☆173Updated last month
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- TbGAL: Tensor-Based Geometric Algebra Library☆27Updated 3 years ago
- Designing bridge trusses with Pytorch autograd☆61Updated last year
- Exponential families for JAX☆73Updated last week
- ☆60Updated 3 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Tools for JAX☆49Updated last week
- Scalable Probabilistic Programming Library☆21Updated 7 years ago
- Turn SymPy expressions into trainable JAX expressions.☆346Updated 3 months ago