List: Jax | Curated by Ebrahim Pichka | Medium

Jan 31, 2023

6 stories

Jax

Khang Pham

Intro to JAX for Machine Learning

JAX is a Python library designed for high-performance numerical computing, especially machine learning research.

Dec 14, 2022

Intro to JAX for Machine Learning

Dec 14, 2022

In

Towards Data Science

by

Cristian Garcia

Deep Learning with Jax and Elegy

Going beyond TensorFlow, Pytorch, and Keras

Aug 31, 2020

Deep Learning with Jax and Elegy

Aug 31, 2020

In

SyncedReview

by

Synced

DeepMind Releases New JAX Libraries for Neural Networks and Reinforcement Learning

DeepMind announced yesterday the release of Haiku and RLax — new JAX libraries designed for neural networks and reinforcement learning…

Feb 21, 2020

DeepMind Releases New JAX Libraries for Neural Networks and Reinforcement Learning

Feb 21, 2020

Derrick Mwiti

JAX and Flax tutorials

Looking for high-performance with deep learning networks? Look no further. JAX is here to help you.

Sep 15, 2022

Sep 15, 2022

In

Towards Data Science

by

Branislav Holländer

JAX: Differentiable Computing by Google

A Revolutionary Machine Learning Framework

Oct 28, 2020

JAX: Differentiable Computing by Google

Oct 28, 2020

In

Towards Data Science

by

Robert Lange

Getting started with JAX (MLPs, CNNs & RNNs)

A Guided Tour Through jit-grad-vmap-scan Wonderland

Dec 27, 2020

Getting started with JAX (MLPs, CNNs & RNNs)

Dec 27, 2020

Ebrahim Pichka

Ebrahim Pichka

Engineering Graduate Student & Research Assistant, interested in ML, and Optimization. https://epichka.com/

Following

Help
Status
About
Careers
Press
Blog
Privacy
Terms
Text to speech
Teams