Add JAX to speeding up section (#21)

* Add JAX to speeding up section

* Update README.md

Co-authored-by: Michael Eliachevitch <[email protected]>
Co-authored-by: Eduardo Rodrigues <[email protected]>

Author: meliache

README.md

@@ -191,11 +191,13 @@ Often, it is not needed anymore to write C++/C routines that get wrapped since t
 | [numba][] | Tight loops are often the slow part of Python code, where this JIT compiles them! |
 | [Pythran][] | Whole scripts |
 | [numpy][] | Expressing your code as array options means you get native-C speeds per sub-expression. |
+| [jax][] | Allows compiling and running NumPy operations on accelerated hardware such as GPU's. Also offers a JIT compiler (less sophisticated than Numba), automatic analytic differentiation (gradient) of Python functions and efficient vectorization. Was developed with the aim of being able to develop efficient machine learning algorithms "from scratch" with NumPy-like code.
 | [NumExpr][] | Single pass "mapper" operations (one input → one output). |
 
 [numba]:   https://numba.pydata.org
 [Pythran]: https://pythran.readthedocs.io/en/latest/
 [numpy]:   http://www.numpy.org
+[jax]:     Https://jax.readthedocs.io/en/latest/notebooks/quickstart.html
 [NumExpr]: https://numexpr.readthedocs.io/en/latest/user_guide.html
 
 ## Binding C/C++ to Python