Skip to content
← Learn JAX

JAX Numerical Computing

JAX

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

This track uses JAX. You can make it your practice mode to browse related content across the site.

25 exercises

Explore freely. Sign in to keep your progress.

Your learning sequence

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

Check yourself

4 questions · not saved while you're signed out

Use these questions to check your understanding and compare explanations. Your score does not affect track completion.

0 / 4 correct

Which of these integers survive a round trip through float32?

Choose one answer to reveal the explanations. Answers are kept only while this page is open.

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