Edifice - 92 neural network architectures for Nx/Axon

[manhvu]

Thank you for sharing—I’m trying to convert some models from Python to Axon/Nx, but it isn’t finished yet.