Skip to content

Commit

Permalink
Add JAX to speeding up section (#21)
Browse files Browse the repository at this point in the history
* Add JAX to speeding up section

* Update README.md

Co-authored-by: Michael Eliachevitch <meliache@uni-bonn.de>
Co-authored-by: Eduardo Rodrigues <eduardo.rodrigues@cern.ch>
  • Loading branch information
3 people committed Oct 7, 2020
1 parent 621fd1c commit 21db985
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 21db985

Please sign in to comment.