JAX (What it is and how to use it in Python) - discu.eu

Linking pages

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
Google Colab https://colab.research.google.com/#scrollTo=Nma_JWh-W-IF 25 comments
GitHub - elixir-nx/nx: Multi-dimensional arrays (tensors) and numerical definitions for Elixir https://github.com/elixir-nx/nx 10 comments
GitHub - pytorch/xla: Enabling PyTorch on Google TPU https://github.com/pytorch/xla 0 comments
Distributed training with JAX & Flax https://www.machinelearningnuggets.com/distributed-training-with-jax-and-flax/ 0 comments
Google Colab https://colab.research.google.com/notebooks/tpu.ipynb 0 comments
NumPy tutorial(Everything you need to know about NumPy with examples and video tutorials) https://www.machinelearningnuggets.com/numpy-tutorial/ 0 comments
The Autodiff Cookbook — JAX documentation https://jax.readthedocs.io/en/latest/notebooks/autodiff_cookbook.html 0 comments
XLA: Optimizing Compiler for Machine Learning | TensorFlow https://www.tensorflow.org/xla 0 comments

Related searches:

Search whole site: site:machinelearningnuggets.com

Search title: JAX (What it is and how to use it in Python)

See how to search.

Submit link to: