JaxSSO

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

Source-to-Source Debuggable Derivatives in Pure Python

A differentiable PDE solving framework for machine learning

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

Main yt repository

Surrogate modeling and optimization for scientific machine learning (SciML)

large population models