RobinKa / jaxgaLinks
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
Sorting:
- Geometric algebra in JAX and numpy☆81Updated 2 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Pytrees + dataclasses ❤️☆62Updated last month
- Python package for Geometric / Clifford Algebra with TensorFlow☆50Updated 2 years ago
- ☆104Updated 11 months ago
- Documentation:☆120Updated 2 years ago
- TbGAL: Tensor-Based Geometric Algebra Library☆27Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆53Updated last month
- Tools for JAX☆47Updated this week
- Path Tracing in JAX☆71Updated 5 years ago
- ☆60Updated 3 years ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- A JAX implementation of stochastic addition.☆14Updated 2 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- A functional training loops library for JAX☆88Updated last year
- ☆80Updated 3 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 5 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆12Updated 4 years ago
- Exponential families for JAX☆69Updated this week
- A symbolically optimized Geometric Algebra library with PyTorch/NumPy/SymPy/etc. compatibility and ganja.js visualization.☆79Updated last week
- Inference Combinators in JAX☆49Updated 2 weeks ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- ☆17Updated 9 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- Mathematical operations for JAX pytrees☆198Updated 5 months ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago