About the Role
We are looking for Machine Learning engineers who are passionate about generative models and creative applications of AI. In particular, we are looking for people who share our mission of open-source machine learning models; people who do not believe AI models should be controlled by a centralized gatekeeper behind a closed wall, but rather be truly open and in control by all. We want highly creative ML engineers who are motivated to push the boundaries of generative models. You will have access to state-of-the-art high performance computing resources and you will be able to work alongside top researchers and engineers to truly make an impact in the fast growing world of generative AI.
Responsibilities:
Lead efforts to drive the design development and production of ML systems, and present the solutions to customers
Work with Research team on developing the next generation of models, where you may assist with areas such as optimization of model training, model tuning, dataset engineering, HPC clusters, tooling, and work on open-source efforts
Be a strategic thought partner for leaders across the organization on driving business impact through machine learning
Work on the Commercial side - productioning generative models, and building the infrastructure to serve them at scale, or work to build customized models for commercial applications
Prototype and productionize model architecture improvements and new features
Qualifications:
3+ years working on machine learning projects, including training, fine tuning and refining models
Experience with Python scientific stack, PyTorch, creating Jupyter/Colab notebooks
Experience with JAX / TPUs / CUDA-level / JavaScript (TensorFlow.js etc) a plus
Ability to communicate machine learning concepts and results effectively through writing and visualization
Experience with training and/or deploying ML models with Amazon AWS (Sagemaker a plus) or Google Cloud
Experience with building interactive web demos that serve generative ML models
Experience with the open-source ML ecosystem (HuggingFace, W&B, etc.)
Experience with Linux and command line tools