KhawajaAbaid / flax-deep-learning Public

Notifications You must be signed in to change notification settings
Fork 0
Star 1

Deep Learning in my new favorite ecosystem of Flax + Jax + Optax!

1 star 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 18 Commits
README.md		README.md
gan.py		gan.py
gpt2.py		gpt2.py
gpt2_helpers.py		gpt2_helpers.py
kapathy_bigram.py		kapathy_bigram.py
linear_regression.py		linear_regression.py
makemore_mlp.ipynb		makemore_mlp.ipynb
mlp.py		mlp.py
mlp2.py		mlp2.py
vae.py		vae.py

Repository files navigation

Flax Deep Learning

Implementations of popular deep learning architectures in flax!

About

Deep Learning in my new favorite ecosystem of Flax + Jax + Optax!

Report repository

Releases

No releases published

Packages

No packages published

Languages