This README is a very short. To learn everything you need to know about this repo, see our full documentation

If you want to start a project just install pdm, init project and install dependencies and of you go. Everything can be found in the documentation.

• Full train and test pipeline 😛. I don't have a GPU for such a task so if anyone would like to lend me a GPU or try to do it themselves and give feedback and results I would be delighted 👌.
• Make documentation better
• Add package and dependency manager pdm
• Check jaxtyping if it time and memory optimize. If yes I think it would be great to add it. I've decided not to use it because it won't provide much readability as I thought, and I don't think we need to provide more input validation in the architecture. However, it doesn't worsen performance, but I'm not fully convinced that it will be a big plus given that we will be dependent on this framework. If anyone has a different opinion pls share maybe I am wrong.
• Make more tests to make coverage test percentage higher 😏
• Test distributed learning
• Add pip deployment
• Add hugging face deployment