jax-reactor

A fast and flexible implementation of Rigid Body Dynamics algorithms and their analytical derivatives

Source-to-Source Debuggable Derivatives in Pure Python

A simple library for creating complex neural networks

Introductions to key concepts in quantum programming, as well as tutorials and implementations from cutting-edge quantum computing research.

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

Tensors and differentiable operations (like TensorFlow) in Rust

Surrogate modeling and optimization for scientific machine learning (SciML)

:microscope: Nano size Theano LSTM module

Automatic Differentiation Library for Computational and Mathematical Engineering

A deep learning framework created from scratch with Python and NumPy