Re implementing, from scratch the important neural network architectures using JAX
Primary LanguageJupyter Notebook