jax-ai-stack

[all]
wandb>=0.21.1
