/CobraML

AutoML Software designed to give users access to a whole plethora of ML models, some trainable on the GPU.

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause


CobraML: ML for everyone

What is it?

CobraML is a python library built on top of the PyTorch framework, designed to give users access to a whole plethora of ML models, which are trainable on the gpu.

  • Just like PyTorch CobraML gives the user full control when designing their model pipeline, from the cost function to the optimizer everything is in the users hand!

  • CobraML recognizes that simplicity is also important to growing the ML community, and thus has model implementations that are very beginner friendly.

  • CobraML does have gpu support, thanks to PyTorch cuda availability.

  • CobraML is completely open source and in the future we are looking forward to community collaboration and input!

Progress

CobraML is in it's extremely early stages with only a few algorithms ready for production, and is thus not ready for release.

How to contribute?

If you wish to contribute to this library, please click on the following link. All help is gladly welcome! https://docs.google.com/forms/d/e/1FAIpQLScv3zBQSVL38dib6_wxe1ABM5eY0rooMDx_gvXTiO7JuHyW4Q/viewform?usp=sf_link