Software Engineer (ML Infra)

About the job

Adamas Knight is recruiting for a groundbreaking AI Lab, backed by some of the biggest names in industry, working on building their own proprietary foundation model within the multi-modal domain - text and vision.

With one of the best compute in industry, they are looking for a ML Infrastructure Engineer to join the team.

The Role

As a ML Infrastructure Engineer, you will be instrumental in designing, building, and optimizing the infrastructure that supports their deep learning models. Working closely with the Research Scientist and Engineers, you will be central to creating robust machine learning pipelines, managing computational resources, and automating workflows, enabling our team to innovate and deploy AI models at scale.

You will:

• Design and Optimize ML Pipelines: Build and maintain end-to-end machine learning pipelines, including data pre-processing, model training, evaluation, and deployment automation.
• Infrastructure Management: Develop and manage scalable cloud-based and/or on-prem infrastructure to support the execution of machine learning experiments and model training (e.g., AWS, GCP, Azure, Kubernetes, Docker).
• Model Deployment: Work closely with AI researchers to ensure seamless deployment of machine learning models into production environments, focusing on scalability, reliability, and performance.
• Automate Workflow and Resource Management: Implement tools and automation scripts to optimize the use of computing resources, including the management of GPU/TPU resources and distributed training infrastructure.
• Monitoring and Scaling: Build monitoring solutions to track performance, usage, and reliability of ML models and infrastructure, ensuring that systems scale rapidly as needed.
• Continuous Improvement: Stay up to date with the latest trends and advancements in machine learning infrastructure and MLOps, and apply them to enhance team productivity and system performance.

Benefits/Perks

Attractive Compensation:Enjoy a competitive salary and the opportunity to invest in your future with equity in the company

Comprehensive Benefits:Access private healthcare, a gym allowance, and catered lunches to support your well-being

Work-Life Balance:Benefit from flexible working hours that fit your lifestyle

At Adamas Knight, we are committed to creating an inclusive culture. We do not discriminate based on race, religion, gender, national origin, sexual orientation, age, veteran status, disability, or any other legally protected status. Diversity is highly valued, and we encourage applicants from all backgrounds to apply.