Skip to content
← All tracks

JAX Numerical Computing

JAX

Linear algebra, decompositions, FFT, convolutions, sparse arrays, polynomial fitting, numerical stability. Scientific computing in JAX.

This track is written for JAX, which isn't the mode you're browsing in.

0 / 25 solved
Continue →
  1. 1. Not solved yet. linalg.solve
  2. 2. Not solved yet. linalg.inv
  3. 3. Not solved yet. slogdet for Stable log|det|
  4. 4. Not solved yet. linalg.norm (Frobenius)
  5. 5. Not solved yet. linalg.matrix_power
  6. 6. Not solved yet. Cholesky Decomposition
  7. 7. Not solved yet. QR Decomposition
  8. 8. Not solved yet. SVD Singular Values
  9. 9. Not solved yet. eigh: Symmetric Eigendecomp
  10. 10. Not solved yet. 1-D FFT Magnitude
  11. 11. Not solved yet. Real FFT (rfft)
  12. 12. Not solved yet. 2-D FFT DC Component
  13. 13. Not solved yet. 1-D Conv via lax.conv_general_dilated
  14. 14. Not solved yet. 2-D Conv via lax.conv_general_dilated
  15. 15. Not solved yet. Conv Padding & Strides
  16. 16. Not solved yet. BCOO from Dense
  17. 17. Not solved yet. Sparse Matrix-Vector Product
  18. 18. Not solved yet. BCOO Round-Trip
  19. 19. Not solved yet. Polynomial Fitting via polyfit
  20. 20. Not solved yet. Polynomial Evaluation via polyval
  21. 21. Not solved yet. Linear Interpolation
  22. 22. Not solved yet. Numerically Stable logsumexp
  23. 23. Not solved yet. Stable log_softmax
  24. 24. Not solved yet. log1p / expm1 for Small Values
  25. 25. Not solved yet. Cumulative Trapezoidal Integration

Check yourself

4 questions · one attempt each

These do not count toward finishing the track. They are here to catch the things that are easy to read past.

0 / 4

Which of these integers survive a round trip through float32? 

jnp.float32(2**24)      # 16777216
jnp.float32(2**24 + 1)  # 16777217
jnp.float32(2**24 + 2)  # 16777218
Question 1 of 4