Challenge: Basics of JAX
Test your understanding through the following exercises.
Perform the following tasks to practice your concepts based on the basics of JAX. The required JAX and NumPy libraries have already been imported for you.
Challenge 1: JAX arrays
Create an integer JAX ...