mat kelcey
mat_kelcey
jax vmap and pmap are so awesome! consider this example i worked through today...i have a dataset of 2600 imgs across 4 hosts; so 5400 imgs per host i.e. shape
Read more
By continuing to use the site, you are consenting to the use of cookies as explained in our Cookie Policy to improve your experience.