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