Machine Learning Performance Engineer

<p>&nbsp;</p> <p>Optiver is a seeking a Machine Learning Performance Engineer to join our team, focusing on a pivotal AI initiative. This role would offer the opportunity to have significant impact across Machine Learning infrastructure, training, and inference challenges to advance our futures trading strategies.</p> <p><strong>What you'll do:</strong></p> <ul> <li>Build scalable and robust training and inference pipelines for deep learning</li> <li>Dive into internals of open-source deep learning frameworks and enhance their functionality</li> <li>Identify and eliminate performance bottlenecks</li> <li>Collaborate closely with researchers and other engineers</li> <li>Develop an in-depth understanding of trading systems</li> </ul> <p><strong>What you’ll get:</strong></p> <p>You’ll join a culture of collaboration and excellence, surrounded by curious thinkers and creative problem-solvers. Motivated by a passion for continuous improvement, you’ll thrive in a supportive, high-performing environment alongside talented colleagues, collectively tackling some of the toughest challenges in the financial markets.</p> <p>In addition, you’ll receive:</p> <ul> <li>The opportunity to work alongside best-in-class professionals from over 40 different countries</li> <li>A highly competitive compensation package</li> <li>Global profit-sharing pool and performance-based bonus structure</li> <li>401(k) match up to 50%</li> <li>Comprehensive health, mental, dental, vision, disability, and life coverage</li> <li>25 paid vacation days alongside market holidays</li> <li>Extensive office perks, including breakfast, lunch and snacks, regular social events, clubs, sporting leagues and more</li> </ul> <p><strong>Who you are:&nbsp;</strong></p> <ul> <li>Strong knowledge of low-level GPU programming with CUDA, including Tensor Cores, cooperative groups, graphs, and warp-level intrinsics</li> <li>Expertise in internals of deep-learning frameworks like PyTorch, JAX, TensorFlow, etc.</li> <li>Deep understanding of computer architecture</li> <li>Experience in C++ and Python</li> </ul> <p><strong>Nice to have:</strong></p> <ul> <li>Experience with JAX ecosystem (XLA, Flax, etc.)</li> <li>Familiarity <span class="fabric-background-color-mark" data-renderer-mark="true" data-background-custom-color="#fedec8">with</span><strong data-renderer-mark="true"> </strong><span class="fabric-background-color-mark" data-renderer-mark="true" data-background-custom-color="#fedec8">GPU libraries a