Software Engineering • Senior

Who you are

You have 5+ years of experience in software engineering, particularly in building and maintaining complex systems. Your expertise in Python and machine learning frameworks like TensorFlow allows you to design and implement distributed training systems effectively. You understand the intricacies of GPU utilization and have experience optimizing performance in large-scale environments.

You thrive in collaborative settings, working closely with product and infrastructure teams to identify customer needs and translate them into technical solutions. Your strong problem-solving skills enable you to tackle challenges in scalable training infrastructure, ensuring reliability and efficiency in model training processes.

What you'll do

As a Senior Software Engineer – Model Training at Baseten, you will design and build the infrastructure that powers the training and fine-tuning of foundation models. You will implement scalable pipelines that facilitate efficient model adaptation for both Baseten and its customers. Your role will involve optimizing GPU utilization to enhance performance and reliability in model training.

You will take ownership of key components of the training stack, ensuring that the systems you build are robust and scalable. Collaborating with cross-functional teams, you will push the boundaries of what is possible in scalable training infrastructure, contributing to the advancement of AI technologies.

What we offer

At Baseten, you will be part of a forward-thinking team dedicated to innovation in AI. We provide a supportive environment that encourages professional growth and development. You will have the opportunity to work on cutting-edge projects that have a significant impact on the AI landscape. We value diversity and inclusion, fostering a workplace where everyone can thrive.