69 Open source projects that are alternatives of or similar to jaxdf

Massively parallel rigidbody physics simulation on accelerator hardware.

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

Bayesian Optimization in JAX

Toy implementations of some popular ML optimizers using Python/JAX

Parallel programming for everyone.

10 differentiable physical simulators built with Taichi differentiable programming (DiffTaichi, ICLR 2020)

A general-purpose probabilistic programming system with programmable inference

Swift for TensorFlow

A common interface for quadrature and numerical integration for the SciML scientific machine learning organization