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]>
hsf-training · Oct 7, 2020 · 21db985 · 21db985
1 parent 621fd1c
commit 21db985
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/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 &rarr; 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