Staff Software Engineer, Machine Learning Infrastructure

Who you are

You have significant experience in owning large technical initiatives and leading small teams — your background includes designing and architecting distributed microservice systems that are robust and scalable. You are familiar with open-source software and have a strong understanding of machine learning frameworks. Your expertise in cloud platforms such as AWS, GCP, and Azure allows you to effectively manage and deploy machine learning models in production environments.

You have a solid foundation in programming languages, particularly Golang, and are comfortable working with relational databases. Your experience with real-time and asynchronous data processing enhances your ability to build efficient and responsive systems. You are passionate about machine learning and its applications, and you thrive in collaborative environments where you can share knowledge and learn from others.

Desirable

Prior experience with TensorFlow, PyTorch, Onnx, Nvidia Triton, and Kubeflow is a plus, as these tools are integral to our machine learning infrastructure. Familiarity with multiple cloud platforms will give you an edge in this role, as we leverage various services to optimize our solutions. Your ability to prototype new training frameworks and productionize solutions at scale will be highly valued.

What you'll do

In this role, you will be essential in contributing to our core ML infrastructure — you will work closely with research teams to design and build our training infrastructure, ensuring that it meets the needs of our users and researchers. Your responsibilities will include prototyping new training frameworks and optimizing model integration infrastructure to enhance performance and scalability.

You will lead initiatives that involve designing and implementing distributed systems that support machine learning workflows. This includes collaborating with cross-functional teams to ensure that our infrastructure is robust and capable of handling large-scale data processing. You will also be responsible for testing and validating the performance of the systems you build, ensuring they meet the high standards expected in the industry.

Your role will involve mentoring junior engineers and sharing your expertise in machine learning and infrastructure design. You will have the opportunity to influence the technical direction of our projects and contribute to the overall success of our machine learning initiatives.

What we offer

At Clarifai, we offer a competitive salary range of $150,000 - $240,000, depending on your relevant experience. You will be part of a dynamic team that is dedicated to pushing the boundaries of AI and machine learning. We foster a culture of innovation and collaboration, where your contributions will have a direct impact on our products and services.

We encourage you to apply even if your experience doesn't match every requirement — we value diverse perspectives and are committed to building a team that reflects a variety of backgrounds and experiences. Join us in transforming how organizations leverage AI to unlock the potential of their data.