Machine Learning Performance Engineer

Salary $150,000 - $250,000
LocationAustin
Employment type Permanent
Discipline
Technology

Job description

Job Title: Machine Learning Performance Engineer

Location: Austin, Texas or New York, New York

Job 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.