We created this visualization of the SAM model that allows you to see the architecture in a interactive manner, along with the code. We made this when trying to implement the SAM model for us to understand it better. Thought I’d share it and some ML folks might find it useful. Please let me know if it did or did not help you! How do you usually go about understanding the model architectures?