academic/awesome-datascience

Add jax

Closed this issue · 2 comments

Jax has been gaining popularity these days, and I think it's good to also introduce it here.
Not quite sure if it should be added under the section The Data Science Toolbox / General Machine Learning Packages or The Data Science Toolbox / Deep Learning Packages though.

@kevin1kevin1k could you make a PR for this?

Sure! And by the way I decided to add more stuff than just jax.