Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Flax is a neural network library for JAX that is designed for flexibility.

This is the official code release for Bayesian Flow Networks.

Sonne tanken ☀️🚘

A plugin to use Nvidia GPU in PySCF package

DSL for probabilistic models specification and probabilistic programming.

High-performance reactive message-passing based Bayesian inference engine

An attack of the ASCAD dataset in 5 minutes

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Python package for Sentential Decision Diagrams (SDD)

This repository holds the code for the NeurIPS 2022 paper, Semantic Probabilistic Layers

Unguard is an insecure cloud-native microservices demo application.

Statistical Rethinking Course for Jan-Mar 2023

Parallel Computing and Scientific Machine Learning (SciML): Methods and Applications (MIT 18.337J/6.338J)

An open source ML system for the end-to-end data science lifecycle