jax

bde is a Python package that brings sampling-based Bayesian Deep Learning to practitioners via the MILE method, combining JAX's speed with scikit-learn's API for tabular supervised learning tasks.

CAX is a high-performance open-source library for cellular automata research, built on JAX. It accelerates simulations up to 2000x and supports discrete and continuous automata in arbitrary dimensions.