Jax is an amazing abstraction for neural nets
— Wojciech Zaremba (@woj_zaremba) April 12, 2019
- Same api as numpy
- it's functional
- only python, and only 10k lines
- it’s fast because it’s on top of XLA
Many people want to switch to Jax. The only obstacle is the lack of long term commitment from Google. @JeffDean @rajatmonga
from Twitter https://twitter.com/woj_zaremba
April 12, 2019 at 11:23AM
via IFTTT
No comments:
Post a Comment