If you are interested in machine learning, check out Gabor Melis's library: https://github.com/melisgl/mgl. It's not an area I'm super familiar with, so I can't speak to it's feature set, but I believe he used it to win a machine learning competition some years ago.
I don't think anyone's written a transformer or diffusion model with it, could be a fun challenge.
I don't think anyone's written a transformer or diffusion model with it, could be a fun challenge.