Re implementing, from scratch the important neural network architectures using JAX as a test to my knowledge in machine learning and being comfortable with designing my own network architectures. Note these are notes that might not be super clear as they require some knowledge in ML already.
PetchMa/deeplearning_fundamentals
Re implementing, from scratch the important neural network architectures using JAX
Jupyter Notebook