jaxns

Deep Learning for humans

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

aka "Bayesian Methods for Hackers": An introduction to Bayesian methods + probabilistic programming with a computation/understanding-first, mathematics-second point of view. All in pure Python ;)

? A ranked list of awesome machine learning Python libraries. Updated weekly.

Convert Machine Learning Code Between Frameworks

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Trax — Deep Learning with Clear Code and Speed

Python code for "Probabilistic Machine learning" book by Kevin Murphy

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

A probabilistic programming language in TensorFlow. Deep generative models, variational inference.