High-Performance Machine Learning Engineer, Selby Jennings

Responsibilities:

• Build Pipelines: Design and develop scalable and robust training and inference pipelines for deep learning models.
• Enhance Frameworks: Dive into the internals of open-source deep learning frameworks (e.g., PyTorch, JAX, TensorFlow) and improve their functionality.
• Optimize Performance: Identify and eliminate performance bottlenecks in machine learning systems, with a focus on high-performance tools and environments.
• Collaborate: Work closely with researchers and engineers to solve performance challenges.
• Understand Systems: Develop a comprehensive understanding of trading systems and their requirements.

Requirements & Skills:

• Deep Learning Frameworks: Expertise in frameworks such as PyTorch, JAX, and TensorFlow.
• High-Performance Tools: Strong experience with CPUs, GPUs, CUDA, and/or JAX ecosystem tools (e.g., XLA, Flax).
• Programming Skills: Proficiency in C++ and Python.
• Computer Architecture: In-depth knowledge of computer architecture principles.