Fresh off the back of our large seed-stage investment round, we are looking for an experienced Machine Learning Engineer to lead our ML work streams, and translate product concepts and requirements into a mixture of: research tasks, proof-of-concept applications, and production-ready features.
Key Responsibilities:
- Developing and refining advanced machine learning models for image generation, with a core focus on maintaining control of the output. Fine-tuning latent diffusion models and training new conditioning models.
- Optimising model performance for speed, quality, and ensuring scalability in a cloud-based environment.
- Creating and maintaining pipelines for automated model training and deployment.
- Assisting backend developers to build and maintain data engineering pipelines for downstream data analysis and training tasks against a data warehouse.
- Collaborating closely with the CTO, product designers, and engineering team to integrate ML models into the core product.
- Conducting research and staying up-to-date with the latest advancements in machine learning and computer vision relevant to our future roadmap and feature ideas.
- Mentoring junior ML engineers and data scientists, fostering a collaborative and innovative team culture.
- Working with the product team to understand user needs, testing and iterating on features to improve the user experience.
Requirements:
- 5+ years of hands-on experience in machine learning, with a focus on deep learning, image generation, and/or computer vision.
- Good understanding of latent diffusion models, LLMs.
- Strong programming skills in Python and experience with ML libraries like TensorFlow or PyTorch.
- Proven experience deploying ML models in production environments, including model optimisation and scaling.
- Knowledge of cloud platforms (AWS, GCP, or Azure) and containerization (Docker, Kubernetes).
- Excellent problem-solving skills and a self-starter attitude.
It would be a bonus for you to have:
- Master's or Ph.D. in Computer Science, Machine Learning, GenAI, or a related field.
- (Extra bonus.) Experience or familiarity with CAD or 3D modelling software (e.g., Blender, 3DS Max).
Core Skills: Machine Learning, Deep Learning, Python, Docker, Data Warehousing
Other Skills: Tensorflow, PyTorch, AWS
Seniority: Senior, Lead