IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)

State Space Models library in JAX

Python implementation of the directed spectrum

An implementation of the tools described in the paper entitled "Graphical-model based estimation and inference for differential privacy"

Repository for the AugmentedPCA Python package.

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, and backpropagation.

TikZ library for drawing Bayesian networks, graphical models and (directed) factor graphs in LaTeX.

Pytorch implementation of Block Neural Autoregressive Flow

Reference implementation of variational sequential Monte Carlo proposed by Naesseth et al. "Variational Sequential Monte Carlo" (2018)

An implementation of piecewise linear time warping for multi-dimensional time series alignment

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Mice Ultrasonic Profile ExTractor

Deconvolution of calcium imaging data

Deep universal probabilistic programming with Python and PyTorch

A generative network for animal vocalizations. For dimensionality reduction, sequencing, clustering, corpus-building, and generating novel 'stimulus spaces'. All with notebook examples using freely…