Search Results - RepositoryStats

98

662

mit

9

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...

jax lax xla flax haiku numpy optax python jupyter tutorial learn-jax deep-learning machine-learning

Created 2021-10-28

33 commits to main branch, last one 2 years ago

use-lax arthurdenner

1

62

unknown

3

React hook to use with lax.js

lax react animation hacktoberfest

This repository has been archived (exclude archived)

Created 2019-03-25

758 commits to main branch, last one about a year ago

FJFormer erfanzar

4

24

apache-2.0

5

paralleled/unparalleled computational with FJFormer

jax lax llm flax numpy easydel

Created 2023-05-31

244 commits to main branch, last one 9 days ago