tianluyuan / sphere
Implementation of FB8, a generalization of the Kent (1982) and Bingham-Mardia (1978) distributions on a sphere
☆16Updated this week
Related projects ⓘ
Alternatives and complementary repositories for sphere
- A differentiation API for PyTorch☆30Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups☆52Updated 2 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated last year
- Documentation:☆120Updated last year
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆25Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆58Updated last year
- ☆78Updated 3 years ago
- ☆66Updated last year
- ☆22Updated 4 years ago
- Bayesian algorithm execution (BAX)☆46Updated 3 years ago
- ☆15Updated 3 years ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- Differentiable and numerically stable implementation of the matrix exponential☆32Updated 4 years ago
- Pytorch implementation of the Power Spherical distribution☆73Updated 3 months ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Implementation of deep implicit attention in PyTorch☆63Updated 3 years ago
- Efficient Householder Transformation in PyTorch☆60Updated 3 years ago
- ☆49Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆41Updated last year
- ☆12Updated 4 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- Layered distributions using FLAX/JAX☆10Updated 3 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- Normalizing Flows in Jax☆105Updated 4 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated last year
- Pytrees + dataclasses ❤️☆61Updated 4 months ago
- VAEs with Lie Group latent space☆97Updated 2 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago