Sr. Machine Learning Engineer (PyTorch)

About the AIA Labs
AIA Labs is an in-house venture at Bridgewater Associates that is focused on using AI (Artificial Intelligence) to generate returns in markets. Its goal is to develop capabilities that enhance Bridgewater’s existing strategies as well as to manage a dedicated AI strategy for clients.  Bridgewater’s AIA Labs is combining a broad set of AI tools (de-optimized tabular data models, reasoning tools, and LLMs) using proprietary techniques developed in-house to build an AI-based Reasoning Engine capable of performing fundamental, systematic investment research.

About the Role

We are seeking a PyTorch Engineer to build, optimize, and deploy ML models at scale. This role is ideal for engineers who have strong PyTorch expertise and a deep understanding of model optimization, with some exposure to low-level performance improvements (e.g., CUDA, Triton, TensorRT).

You will work closely with researchers to implement state-of-the-art models, accelerate training and inference, and optimize models for efficiency. While this is not a compiler engineering role, a solid grasp of PyTorch internals and performance profiling is a strong plus.

You Will Drive the Following Responsibilities

• Model Implementation & Optimization: Develop and refine deep learning models in PyTorch, ensuring efficiency and scalability.
• Inference Acceleration: Optimize models for deployment using tools like TorchScript, ONNX, TensorRT, Triton, and torch.compile.
• Performance Profiling & Debugging: Identify and resolve bottlenecks using torch.profiler, nvprof, Nsight, and memory profiling.
• Custom Layers & Autograd: Build custom PyTorch modules and implement efficient backward passes when necessary.
• Scalability & Deployment: Work with the ML Ops team to scale models for distributed training and inference.
• Collaboration with Researchers: Translate research insights into robust, high-performance ML systems.

Required Qualifications

• 5+ years’ experience in a technically rigorous/elite environment with modern full stack tech stacks.
• Strong PyTorch expertise (torch.nn, torch.autograd, torch.compile, torch.fx).
  • Solid understanding of model architectures (e.g., Transformers, CNNs) and deep learning fundamentals.
  • Experience with model optimization (e.g., mixed precision, checkpointing, memory-efficient execution).
  • Ability to profile and debug PyTorch performance (e.g., CUDA memory issues, kernel execution time).
• Strong engineering background (Python, structured software development, scalable ML workflows).

Nice to Have Qualifications

• Some familiarity with CUDA kernel development.
  • Experience with Triton or TensorRT for inference acceleration.
  • Understanding of distributed training frameworks (e.g., DeepSpeed, FSDP, Horovod).
  • Familiarity with PyTorch compilation (torch.compile, torch.fx, MLIR optimizations).

Location Requirements

The anticipated onsite requirement for this role is four days per week at our New York City campus.

Compensation Band

The compensation range for this role is $500,000 to $800,000 inclusive of base salary, discretionary bonus, and long-term incentive. The expected base salary for this role is between 40-66% of the range.

Bridgewater Employee Benefits

One of our core priorities at Bridgewater is to enable our employees to build a great life and career, and we believe our benefits are an important extension of that philosophy. As such, currently Bridgewater offers a competitive suite of benefits including:

• Health insurance with 100% premium covered and access to additional concierge medical services
• 401(k) plan with generous employer match
• Paid time off, including fully paid parental leave and a competitive PTO package
• Workplace flexibility and access to back up childcare
• Financial assistance for family building support, including adoption and egg freezing
• Workplace wellness, including on-site gyms, free meals and healthy snacks, and meditation rooms
• An engaged and active community that includes many company events, affinity networks, and extracurricular interest groups

Bridgewater reserves the right to change its current benefits program at any time, in a manner that is consistent with applicable federal and state regulations.

About Bridgewater

Bridgewater Associates is a premier asset management firm, focused on delivering unique insight and partnership for the most sophisticated global institutional investors. Our investment process is driven by a tireless pursuit to understand how the world’s markets and economies work — using cutting edge technology to validate and execute on timeless and universal investment principles.

Founded in 1975, we are a community of independent thinkers who share a commitment for excellence. By fostering a culture of openness, transparency, and inclusion, we strive to unlock the most complex questions in investment strategy, management, and corporate culture.

Explore more information about Bridgewater on our website here.

Our Culture

Our culture is tied to our mission, which is to understand how the world’s markets and economies work, and to translate that into world-class investment performance. This requires an extraordinary team, defined by the best individuals and the best portfolio of people, operating in an idea meritocracy. In this, we are uncompromising on our shared values of excellence, truth, integrity, humility, and courage. Our aspiration of a vibrant idea meritocracy requires diversity of thought – this comes from diversity of background, identity, experience, and so much more. We pursue inclusion because it’s how we unlock the power of that diversity and make sure the best ideas get heard. To sustain this way of being through lots of years and lots of circumstances, we invest in meaningful work and meaningful relationships – the purpose and joy that comes from tackling something challenging and worthy, alongside people who make us better versions of ourselves.

Explore more information about Bridgewater’s culture on our website here.

This job description is not a contract and confers no contractual rights, privileges, or benefits on any applicant or potential applicant. Bridgewater has the right to change any and all terms of this job description, including, but not limited to, job responsibilities, qualifications and benefits. Nothing in this job description constitutes an offer or guarantee of employment. Please note that we provide immigration sponsorship for this position.

Bridgewater Associates, LP is an Equal Opportunity Employer