NumPy & JAX

Hands-on array programming: broadcasting, indexing, vectorization, einsum, and the JAX numpy API (functional updates, vmap/grad/jit in practice, PRNG keys, pytrees).

0 / 6 solved
0/6 solved
Question Tags Difficulty Success
1. Broadcast Compatibility
numpybroadcasting
Easy 40%
2. View or Copy
numpymemory-layout
Easy —
3. Keepdims Shape
numpyreductions
Medium —
4. Batched Einsum
numpyeinsum
Medium —
5. Functional Update
jaximmutability
Medium —
6. vmap + PRNG Keys
jaxvmap +1
Hard —

Up next

6 of 6 questions here are still unsolved.