3 ms·
What you are saying is fascinating. Unfortunately, my google powers are limited today. All I can see about program transformations in Jax is [1], and it appear
by credit_guy 4y ago
What you are saying is fascinating. Unfortunately, my google powers are limited today.
All I can see about program transformations in Jax is [1], and it appears to me there are 4, grad, jit, vmap and pmap. It seems to me you are implying that there are ways to create custom transformations, and this has actual use cases.
Would you mind giving some more details? Or maybe some links? I can't help but be excited by your enthusiasm, it feels like Jax could be the ultimate programming language.
[1] https://github.com/google/jax#transformations https://github.com/google/jax#transformations
- mccoyb 4y agoI would start here: https://jax.readthedocs.io/en/latest/notebooks/Writing_custom_interpreters_in_Jax.html https://jax.readthedocs.io/en/latest/notebooks/Writing_custo... Note that JAX enforces certain limitations, so you should be careful when considering JAX to be "the perfect language" - in general I don't think this is true. It's quite good at what it's designed for.
- credit_guy 4y agoThank you