AI Research Engineer, Pre-Training

Wehrtyou; Greater London; Sur place; GBP 188 000 - 225 000; Plein temps

Résumé du poste

Hudson River Trading (HRT) is seeking an AI Research Engineer (Pre-training) to join HAIL, the AI Labs responsible for developing foundation models for markets. You will improve large-scale model training, data loading, parallelism, and fault tolerance while collaborating with researchers to shape the research agenda.

The role involves work on multiple GPU clusters, with a strong emphasis on impactful system design and ongoing innovation.

Qualifications

• High engineering skill in GPU kernel and base DL stack development.
• At least two years building deep learning systems across domains.
• Ability to translate methods between application areas is valued.
• LLM experience is valuable but not required.
• Finance experience is not required.

Responsabilités

• Improve all aspects of large-scale model training, including kernel development, data loading, parallelism, networking and fault-tolerance.
• Collaborate with researchers to co-design and improve models and shape research directions.
• Work with multiple GPU clusters to maximize performance and efficiency.

Connaissances

CUDA kernel development

Lower-level DL frameworks

CUDA Graphs

FPGA/ASIC experience

LLM experience (valued)

Description du poste

Hudson River Trading (HRT) is seeking an AI Research Engineer (Pre-training) to join the HAIL team. HAIL (HRT AI Labs) is the team at HRT responsible for developing and maintaining our most powerful models, which are used by our trading teams to drive a significant fraction of our trading. We are building and deploying "foundation models for markets", that ingest and train on vast amounts of market and “alternative” data (such as language) to make predictions about future market state.

As a pre-training research engineer on HAIL, you will have a general mandate to improve all aspects of large-scale model training, including but not limited to kernel development, training data loading, parallelism, networking, and fault-tolerance. You will work closely with our researchers to co-design and improve our models, and shape the research agenda.

We have multiple large, modern, and rapidly growing GPU clusters, and we maintain a very high GPU-to-researcher ratio. We are simultaneously pursuing multiple strategies and developing many model types with different purposes, and we are strongly incentivized to squeeze as much as we can out of our systems. Your work will be directly, clearly, and highly impactful on the business, and it will be challenging: this is a field with no easy or obvious solutions.

Qualifications

• Strong engineering skills, especially any of: CUDA/Triton/Pallas/CuTe DSL kernel development, lower-level PyTorch/JAX/XLA development, CUDA Graphs, FPGA/ASIC experience
• Must have two or more years work experience building deep learning systems, for any domain: robotics, biology, chemistry, physics, audio, video, recommendations, etc.
• Experience translating methods between areas of application is highly valued
• LLM experience is valuable, but not necessary
• Finance experience is not required

The estimated base salary range for this position is 250,000 to 300,000 USD per year (or local equivalent). The base pay offered may vary depending on multiple individualized factors, including location, job-related knowledge, skills, and experience. This role will also be eligible for discretionary performance-based bonuses and a competitive benefits package.

Culture

Hudson River Trading (HRT) brings a scientific approach to trading financial products. We have built one of the world's most sophisticated computing environments for research and development. Our researchers are at the forefront of innovation in the world of algorithmic trading.

At HRT we welcome a variety of expertise: mathematics and computer science, physics and engineering, media and tech. We’re a community of self-starters who are motivated by the excitement of being at the cutting edge of automation in every part of our organization—from trading, to business operations, to recruiting and beyond. We value openness and transparency, and celebrate great ideas from HRT veterans and new hires alike. At HRT we’re friends and colleagues – whether we are sharing a meal, playing the latest board game, or writing elegant code. We embrace a culture of togetherness that extends far beyond the walls of our office.

Feel like you belong at HRT? Our goal is to find the best people and bring them together to do great work in a place where everyone is valued. HRT is proud of our diverse staff; we have offices all over the globe and benefit from our varied and unique perspectives. HRT is an equal opportunity employer; so whoever you are we’d love to get to know you.

Please be advised: Use of AI tools during interviews or assessments is strictly prohibited, unless otherwise instructed or agreed upon. We employ various methods to evaluate the authenticity of candidate responses. If we determine that AI assistance was used during any stage of the hiring process, we reserve the right to immediately disqualify your candidacy or rescind any job offers extended.