Neil Girdhar

Also, is there an asymmetry between `vmap` and `split`? Currently, if you split a key array `k`: ```python h = split(split(k, 3), 5) ``` then `h.shape` is `(3, 5)`. But...

> My plan was for numpy split to be inapplicable altogether, Yes, great plan! I was just confused by the different result. Now I see that it's a numpy function...

> So long as any amount of numpy is supported, I'm wary of possible ambiguity. Yes, very good point. > I otherwise appreciate your case in favor of it. We...

Sorry about the links. I just reorganized the repo a few minutes ago. > Not 100% sure what you are referring to. I'll explain what I was getting at. Currently,...

> I think you're essentially describing an interface similar to what I had in mind for the re-design of this API. The main idea is to no longer require two...

I agree with @manuel-delverme. I also follow PEP8 except for the 80 character limit. I'm using 100 characters now. Now that I try to use type annotations, I find 80...

[off topic]

> It seems it's very specific to jax It's not specific to Jax at all. That was just an example I gave. I'm sure you can find [many, many code...

Thanks Pierre 😄

> Is there an actual performance benefit? Or is this just style? It's motivated by style (I haven't tested the performance). I got the idea from the Googe style guide,...