tianluyuan / sphere
Implementation of FB8, a generalization of the Kent (1982) and Bingham-Mardia (1978) distributions on a sphere
☆16Updated last week
Related projects ⓘ
Alternatives and complementary repositories for sphere
- ☆22Updated 3 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- ☆78Updated 3 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- Documentation:☆120Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- ☆22Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- A LinearOperator implementation for PyTorch☆18Updated 3 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆64Updated last month
- ☆99Updated 4 months ago
- 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
- B-Spline CNNs on Lie groups☆53Updated 4 years ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 2 years ago
- Pytrees + dataclasses ❤️☆64Updated 4 months ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- The differentiable OSQP solver layer for PyTorch☆59Updated last year
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- Tree Approximate Message Passing☆29Updated 6 months ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups☆52Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- ☆66Updated last year
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 5 years ago
- All things Monte Carlo, written in JAX.☆30Updated 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
- Normalizing Flows in Jax☆105Updated 4 years ago
- ☆15Updated 3 years ago