- Notifications
You must be signed in to change notification settings - Fork0
A pathway and collection of resources to learning Jax from beginning to advance.
License
deterministic-algorithms-lab/Jax-Journey
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Blog :Getting Started with Jax by Robert Tjarko Lange
Notebook :jax_basic.ipynb
Blog :Functional JS 2 ,Functional JS 3 by Krzysztof Czernek
A nice comment on Jax :jax-ml/jax#196 (comment)
Blog :From PyTorch to Jax : Towards NN frameworks that purify stateful code by Sabrina J. Mielke
Notebook :making_functions_pure.ipynb
Blog :Documentation Tutorial
Notebook :Basics-MLP ,Basics-MNIST
Note : Now is the best time to readJax-The Sharp Bits
Notebook :haiku basics with lstm
Blog :From PyTorch to Jax : Towards NN frameworks that purify stateful code by Sabrina J. Mielke
Notebook :basic_transformer.ipynb
Blog :FineTuning Transformers with Jax+Haiku by Madison May
The following tutorials form a nice starting point, for a deeper dive into Jax mechanics :
About
A pathway and collection of resources to learning Jax from beginning to advance.
Topics
Resources
License
Uh oh!
There was an error while loading.Please reload this page.