Machine Learning Performance Engineer

Job Title: Machine Learning Performance EngineerLocation: Austin, Texas or New York, New YorkJob Description: We are seeking a Machine Learning Performance Engineer to join our team, focusing on a pivotal AI initiative. This role offers the opportunity to significantly impact Machine Learning infrastructure, training, and inference challenges to advance futures trading strategies.Responsibilities:Build scalable and robust training and inference pipelines for deep learning. Enhance the functionality of open-source deep learning frameworks. Identify and eliminate performance bottlenecks. Collaborate closely with researchers and other engineers. Develop an in-depth understanding of trading systems.Qualifications:Expertise in the internals of deep-learning frameworks like PyTorch, JAX, TensorFlow, etc. Deep understanding of computer architecture. Experience programming in C++ and Python.Preferred Qualifications:Experience with the JAX ecosystem (XLA, Flax, etc.). Experience in programming for GPUs or other accelerators (CUDA, Triton, Pallas, etc.). Linux system programming experience. Experience with large-scale distributed training. Contributions to open-source projects related to data science and machine learning.Benefits:Work alongside best-in-class professionals from over 40 different countries. Performance-based bonus structure that is unmatched in the industry, fostering a truly collaborative environment. Ownership over initiatives that directly solve business problems. An ideal location in a vibrant city with a strong focus on innovation and a high quality of life.Discipline(s): TechnologyDuration: PermanentJob type: Permanent