jax-ml/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (opens in new tab)

Covered by 3 sources including Google Cloud Blog, metaworld.meDiscussed on Hacker News

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more - jax-ml/jax

Read the original article

Sign in to keep reading the full article.

Covered in 4 articles

Google Cloud Blog·

How we evolved Google’s global and data center networks for the AI era

Google Cloud Blog·

How we evolved Google’s global and data center networks for the AI era

A Case for Tracing Based DSL Kernel Languages

Discussed on Hacker News