Machine Learning Engineer

Machine Learning Engineer.

Machine Learning Engineer



$150000 - $200000 per year




Machine Learning (ML)

Join our pioneering GenAI startup, where we are dedicated to creating cutting-edge AI infrastructure for both open source and enterprise solutions. Our mission is to empower developers and organizations by providing advanced tools and frameworks that accelerate AI development and deployment.

Role Overview: We are seeking a highly skilled Machine Learning Engineer specializing in GPU kernels and compilers. In this role, you will be at the forefront of optimizing and developing high-performance computational solutions that are integral to our AI infrastructure. You will work closely with our innovative team to enhance the efficiency and scalability of AI applications.

Key Responsibilities:

  • Develop and optimize GPU kernels for machine learning and deep learning models.
  • Design and implement advanced compiler techniques to improve the performance of AI algorithms.
  • Collaborate with cross-functional teams to integrate and deploy high-performance solutions.
  • Contribute to the development of open-source AI tools and frameworks.
  • Stay updated with the latest advancements in GPU computing, compilers, and AI technologies.


  • Bachelor's, Master's, or Ph.D. in Computer Science, Electrical Engineering, or a related field.
  • Proven experience in GPU programming (CUDA, OpenCL) and performance optimization.
  • Strong understanding of compiler design and implementation.
  • Proficiency in programming languages such as C++, Python, and experience with ML frameworks like TensorFlow or PyTorch.
  • Familiarity with AI and machine learning algorithms and their practical applications.
  • Excellent problem-solving skills and the ability to work in a fast-paced, collaborative environment.

What We Offer:

  • Competitive salary and equity options.
  • Flexible working hours and remote work opportunities.
  • The chance to work with a passionate and innovative team.
  • Opportunities for professional growth and development.
  • The ability to make a significant impact on the future of AI infrastructure.

How to Apply: If you are excited about the potential of AI and have a passion for GPU computing and compiler development, we would love to hear from you!

Join us in revolutionizing AI infrastructure and making a difference in the open source and enterprise landscapes!

Darwin Recruitment is acting as an Employment Agency in relation to this vacancy.

George Wiles


Max. file size: 512 MB.


This field is for validation purposes and should be left unchanged.


Group 365

Are you looking for a new role? Would you like to compare your current salary against the market rate?

Maybe you’re looking to grow your team and need help planning and setting hiring budgets.

Our interactive market update, split by technology, has all the information you need; from salary benchmarks, gender split and average tenure to 'time to hire' and fastest-growing skills.