GitHub repository: mfschubert/agjax
Primary language: Python
Description: Jax wrapper for autograd-differentiable functions