get-started-with-JAX icon indicating copy to clipboard operation
get-started-with-JAX copied to clipboard

verified publisher icon gordicaleksa
→

Metadata

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as t...

Results 2 get-started-with-JAX issues
Sort by recently updated
recently updated
newest added
trafficstars

Minor Change (Spelling): Correct spelling of parameter

jay-tau avatar jay-tau

Replace jax.tree_util.tree_map() with jax.tree_util.tree_multimap()

`jax.tree_util.tree_multimap()` has been removed in JAX 0.3.16. `jax.tree_util.tree_map()` is a direct replacement. `jax.tree_map()` also works instead of `jax.tree_util.tree_multimap()`.

oikosohn avatar oikosohn

← Metadata

578
Stars
92
Forks
Watchers

Owner

verified publisher icon gordicaleksa

Metadata

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as t...

Back