Robert Lange

PhD Fellow@TU Berlin. Intelligence. Biological&Artificial { 🐠 ♥️ 🦈}

Getting started with JAX (MLPs, CNNs & RNNs)
Learn the building blocks of JAX and use them to build some standard Deep Learning architectures (MLP, CNN, RNN, etc.).
jax xla autograd tpu