ML Application Simulator Software Engineer

Who you are

You have a strong background in software engineering with a focus on machine learning applications — your experience includes designing and developing complex systems that model ML workloads. You are proficient in Python and have hands-on experience with frameworks like TensorFlow and Keras, enabling you to create effective simulations that provide actionable insights into system performance. You understand the intricacies of ML workloads and can translate them into software solutions that guide architectural optimizations.

You thrive in collaborative environments, working closely with performance engineers and ML researchers — your ability to communicate complex ideas clearly helps bridge the gap between different teams. You are committed to continuous learning and adapting your skills to keep pace with the evolving landscape of AI technologies. You have a passion for problem-solving and enjoy tackling challenging technical issues.

Desirable

Experience with performance analysis tools and techniques is a plus, as is familiarity with silicon architecture. You may also have a background in working with large-scale data sets and optimizing algorithms for performance. A proactive approach to identifying potential improvements and a willingness to experiment with new technologies will set you apart.

What you'll do

In this role, you will design, develop, and maintain an application-level ML simulator that models end-to-end ML workloads — your work will provide critical insights into system-level performance and inform architectural decisions. You will collaborate closely with the ML Software Performance Analysis team, ensuring that the simulator evolves alongside real workloads and supports key decision-making across the organization. Your contributions will directly impact the development of new products and the optimization of existing systems.

You will engage in regular discussions with performance engineers to understand the requirements and constraints of the systems you are simulating — this collaboration will help you refine the simulator's capabilities and ensure it meets the needs of various stakeholders. You will also work with ML researchers to incorporate the latest advancements in machine learning into the simulator, keeping it relevant and effective.

Your role will involve testing and validating the simulator's performance against real-world scenarios — you will analyze results and iterate on the design to enhance accuracy and usability. You will document your findings and share insights with the broader team, contributing to a culture of knowledge sharing and continuous improvement.

What we offer

Graphcore provides a dynamic work environment where innovation is encouraged — you will have the opportunity to work alongside some of the brightest minds in the industry, contributing to cutting-edge technology in the AI space. We offer a competitive salary range of PLN 303,800 - 411,000, reflecting your experience and the value you bring to the team. Our commitment to inclusivity means that we welcome diverse perspectives and experiences, fostering a collaborative atmosphere where everyone can thrive.

We understand the importance of work-life balance and offer flexible working arrangements to accommodate your needs. You will have access to ongoing professional development opportunities, allowing you to grow your skills and advance your career within the company. Join us at Graphcore and be part of a team that is shaping the future of AI compute.