← Back to context Comment by semiinfinitely 11 hours ago all issues raised are addressed by jax.vmap 2 comments semiinfinitely Reply patrickkidger 11 hours ago Went scrolling looking for this! Most of the article is about problems solved in JAX.Also worth noting the Array API standard exists now. This is generally also trying to straighten out the sharp edges. amensch 9 hours ago Same here, beautiful solution
patrickkidger 11 hours ago Went scrolling looking for this! Most of the article is about problems solved in JAX.Also worth noting the Array API standard exists now. This is generally also trying to straighten out the sharp edges. amensch 9 hours ago Same here, beautiful solution
Went scrolling looking for this! Most of the article is about problems solved in JAX.
Also worth noting the Array API standard exists now. This is generally also trying to straighten out the sharp edges.
Same here, beautiful solution