MHN_minimal Minimal implementation of Modern Hopfield Networks in PyTorch Requirements : PyTorch, Torchvision, Numpy and Matplotlib Running main.py should produce the following figure:
