Joshuaalbert / jaxnldsView external linksLinks
Inference on non-linear dynamical systems written in JAX
☆11Aug 20, 2020Updated 5 years ago
Alternatives and similar repositories for jaxnlds
Users that are interested in jaxnlds are comparing it to the libraries listed below
Sorting:
- Jax-based MaxEnt☆17Nov 24, 2019Updated 6 years ago
- Code for the Spyndra Robot☆11Nov 3, 2018Updated 7 years ago
- Probabilistic Solution of Differential Equations☆13Jun 19, 2022Updated 3 years ago
- Python 3.6 and TensorFlow implementation of the AReS and MaRS algorithms☆11Jun 23, 2019Updated 6 years ago
- Automatic Differentiation for high-performance stencil loops☆13Mar 25, 2021Updated 4 years ago
- Recognizing and exploiting conjugacy without a domain-specific language☆36Dec 1, 2019Updated 6 years ago
- ☆30Feb 11, 2021Updated 5 years ago
- Procedural Generation Using Differentiable Programming☆16Sep 29, 2019Updated 6 years ago
- ☆16Jun 3, 2025Updated 8 months ago
- Tree-structured recurrent switching linear dynamical systems☆38Jul 13, 2020Updated 5 years ago
- [NeurIPS 2020] Simple and practical private mean and covariance estimation.☆35Oct 4, 2020Updated 5 years ago
- An adaptive training algorithm for residual network☆17Aug 22, 2020Updated 5 years ago
- Julia implementation of Pullback Bundle Dynamical Systems (PBDS) framework for geometrically-consistent, multi-task reactive control on m…☆19Sep 17, 2021Updated 4 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆15Jan 18, 2019Updated 7 years ago
- ☆18Mar 20, 2022Updated 3 years ago
- Compiler with automatic differentiation☆49Oct 18, 2023Updated 2 years ago
- ☆18May 17, 2023Updated 2 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Nov 9, 2019Updated 6 years ago
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- Environments to support https://github.com/sholtodouglas/learning_from_play and reinforcement learning for robotic manipulation.☆21Mar 28, 2021Updated 4 years ago
- PhD Publications and Thesis on LASSO Model Predictive Control☆20Jun 2, 2019Updated 6 years ago
- The Limited Multi-Label Projection Layer☆59Jul 25, 2024Updated last year
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆29Nov 26, 2022Updated 3 years ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆23Jul 15, 2022Updated 3 years ago
- Port-Hamiltonian Approach to Neural Network Training☆27Dec 11, 2019Updated 6 years ago
- Algebraic proof discovery in Agda☆35Dec 6, 2021Updated 4 years ago
- Implementation of "Fast and Flexible Temporal Point Processes with Triangular Maps" (Oral @ NeurIPS 2020)☆23Dec 1, 2023Updated 2 years ago
- Object-Oriented Programming☆12Aug 26, 2021Updated 4 years ago
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27May 13, 2019Updated 6 years ago
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆27Oct 16, 2020Updated 5 years ago
- Code for controlling a Kinova Gen3 Manipulator via Drake.☆32Sep 20, 2023Updated 2 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Feb 3, 2020Updated 6 years ago
- python package implementing a multivariate Horner scheme for efficiently evaluating multivariate polynomials☆32Mar 8, 2025Updated 11 months ago
- ☆30Feb 11, 2022Updated 4 years ago
- Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.☆27Aug 9, 2024Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆29Sep 25, 2021Updated 4 years ago
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆70Jul 24, 2022Updated 3 years ago
- ☆26Apr 9, 2019Updated 6 years ago
- Code for density estimation with nonparametric cluster shapes.☆39Jun 8, 2016Updated 9 years ago