Numpy on the GPU? Speeding up Simple Machine Learning Algorithms with JAX

preview_player
Показать описание
Description
#fancyfueko #jax #python

Introduction (0:00)
Python APIs for Machine Learning (0:56)
The Benefits of Functional Programming Languages (2:30)
scikit-learn vs. JAX (5:08)
PyTorch vs. JAX (7:24)
Conclusion (9:17)
Related Videos + Subscribe (9:26)

Part 1: Fast, functional, & differentiable languages/libraries - Building blocks for deep learning

Check out Tom's Channel:

Code for this video:

If you like this video, subscribe to my channel:

Follow me on IG: @fancyfueko
Рекомендации по теме
Комментарии
Автор

anyone interested in trying out JAX (or better yet, a functional approach to parallelization)?

fancyfueko
Автор

Damn, I could have used this 4 weeks ago for an actual problem. Instead I calculated the derivatives the pedestrian way with pen and paper and wrote them in numba. Could've even used the GPU and saved myself some waiting.

xelaxander