Add JAX practice

[danibcorr]

Proposal Summary

The idea is to include an exercise with JAX, so that students can learn about this additional Python library. We can add a slide presenting the syntax.

Motivation

Provide information about alternatives to existing libraries, advantages, and disadvantages. Possibly refer to Gemini 3 and what it entails.

Open Source Status

• Implementation code is available

Useful Links & References

No response

[danibcorr]

Maybe using Jax for TPUs in Google Colab