Случайно попался мне на глаза, пожалуй, лучший манул по JAX, который покрывает множество тем от базовых вещей до более продвинутых вроде multi-node multi-GPU обучения.
https://www.jan-ebert.com/blog/jax-intro.html
Post #106
2.41K
Forwarded from Mamkin Machinist
Jan-Ebert JAX is for Joy, AutoDiff, and Xeleration – Jan Ebert ♥ Software engineer Jan Ebert's homepage containing a blog and web applications. Dealing with performance, machine learning, simulations, mathematics and programming languages like Rust, Julia, C, Lisp, Assembly and Python.- 🔥 13
- 🤝 1