play_with_jax. My attempt to learn jax.

github.com/ruotianluo/play_with_jax
by @ruotianluo

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.