caujax (pypi): Causal Jax
Keywords: Causal|Inference|Jax|Machine|Learning