hanyas

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Kalman-and-Bayesian-Filters-in-Python

Kalman Filter book using Jupyter Notebook. Focuses on building intuition and experience, not formal proofs. Includes Kalman filters,extended Kalman filters, unscented Kalman filters, particle filters, and more. All exercises include solutions.

mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.

annotated_latex_equations

Examples of how to create colorful, annotated equations in Latex using Tikz.

drake

Model-based design and verification for robotics.

user.js

user.js -- Firefox configuration hardening

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

hasktorch

Tensors and neural networks in Haskell

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

bibtex-tidy

Cleaner and Formatter for BibTeX files

chex

Cthulhu.jl

The slow descent into madness

Dojo.jl

A differentiable physics engine for robotics

RigidBodyDynamics.jl

Julia implementation of various rigid body dynamics and kinematics algorithms

trajax

SDE

Example codes for the book Applied Stochastic Differential Equations

posteriordb

Database with posteriors of interest for Bayesian inference

Birch

A probabilistic programming language that combines automatic differentiation, automatic marginalization, and automatic conditioning within Monte Carlo methods.

SGMCMCJax

Lightweight library of stochastic gradient MCMC algorithms written in JAX.

diffarticulated

Efficient Differentiable Simulation of Articulated Bodies (ICML2021)

CALIPSO.jl

Conic Augmented Lagrangian Interior-Point SOlver

logreg

Bayesian inference for a logistic regression model in various languages

dissertation

Zheng Zhao's doctoral dissertation from Aalto University

ilqc

Iterative Linearized Control Toolbox

sequential-parallelization-examples

This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.

corati_home

A collection of user models and parameter estimation methods.

value_expansion

sqrt-parallel-smoothers

A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algorithms, both in a sequential and parallel fashion, as well as efficient gradient rules for computing gradients of required quantities (such as the pseudo-loglikelihood of the system).

aux-ssm-samplers

Auxiliary Kalman and particle Gibbs samplers for Feynman-Kac (state-space) models.

jax-workshop

JAX workshop for Simo Särkkä's group