Closed dscolby closed 6 months ago
Since we are going to be switching roles and I implemented a lesser known model ie not just an MLP, CNN, RNN, etc and it is in Jax, I need to record a quick overview of the model architecture and how to train it.
Done
Since we are going to be switching roles and I implemented a lesser known model ie not just an MLP, CNN, RNN, etc and it is in Jax, I need to record a quick overview of the model architecture and how to train it.