From PyTorch to JAX: towards neural net frameworks that purify stateful code — Sabrina J. Mielke - discu.eu

Linking pages

Implementing Graph Neural Networks with JAX http://gcucurull.github.io/deep-learning/2020/04/20/jax-graph-neural-networks/ 0 comments
JAX Ecosystem. JAX by Google Research is getting more… | by Grigory Sapunov | Medium https://moocaholic.medium.com/jax-a13e83f49897 0 comments
Tutorial: Writing JAX-like code in PyTorch with functorch - Simone Scardapane https://www.sscardapane.it/tutorials/functorch/ 0 comments

Linked pages

GitHub - google/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more https://github.com/google/jax 99 comments

Related searches:

Search whole site: site:sjmielke.com

Search title: From PyTorch to JAX: towards neural net frameworks that purify stateful code — Sabrina J. Mielke

See how to search.

Submit link to: