Machine Learning Performance Engineer
Company Summary: A leading tech-driven trading firm is seeking a Machine Learning Performance Engineer to join a pivotal AI initiative in New York City. This role offers the opportunity to make a significant impact on machine learning infrastructure, training, and inference challenges that directly support advanced trading strategies.
Key Responsibilities
- Build scalable and robust training and inference pipelines for deep learning models.
- Dive into the internals of open-source deep learning frameworks and enhance their functionality.
- Identify and eliminate performance bottlenecks across systems.
- Collaborate closely with researchers and engineers to optimize workflows.
- Develop a deep understanding of trading systems and their integration with ML solutions.
Key Qualifications
- Expertise in the internals of deep learning frameworks such as PyTorch, JAX, TensorFlow, etc.
- Strong understanding of computer architecture.
- Proficiency in C++ and Python programming.
Nice to Have
- Experience with the JAX ecosystem (XLA, Flax, etc.).
- Programming experience for GPUs or other accelerators (CUDA, Triton, Pallas, etc.).
- Linux system programming skills.
- Familiarity with large-scale distributed training.
- Contributions to open-source projects in data science or machine learning.
What You'll Get
- Work alongside top-tier professionals from over 40 countries.
- A performance-based bonus structure that fosters collaboration and rewards impact.
- Ownership of initiatives that directly solve business problems.
- Generous benefits including:
- 25 paid vacation days plus market holidays
- Daily breakfast and lunch
- Training and development opportunities
- Charitable match programs
- Regular social events and clubs
FAQs
Congratulations, we understand that taking the time to apply is a big step. When you apply, your details go directly to the consultant who is sourcing talent. Due to demand, we may not get back to all applicants that have applied. However, we always keep your resume and details on file so when we see similar roles or see skillsets that drive growth in organizations, we will always reach out to discuss opportunities.
Yes. Even if this role isn’t a perfect match, applying allows us to understand your expertise and ambitions, ensuring you're on our radar for the right opportunity when it arises.
We also work in several ways, firstly we advertise our roles available on our site, however, often due to confidentiality we may not post all. We also work with clients who are more focused on skills and understanding what is required to future-proof their business.
That's why we recommend registering your resume so you can be considered for roles that have yet to be created.
Yes, we help with resume and interview preparation. From customized support on how to optimize your resume to interview preparation and compensation negotiations, we advocate for you throughout your next career move.