About the Role
Our ML team sits at the intersection of research, engineering, and platform development. We partner closely with researchers and investors to prototype novel modeling ideas, build high-performance ML infrastructure, and productionize models that run at scale across massive datasets. You’ll play a key role in both engineering ML models and building the platforms and pipelines that make research faster and production systems more robust.
Responsibilities
- Prototype and experiment with ML models across a range of techniques, from traditional ML methods to deep learning architectures.
- Engineer promising research models for production, including data pipelines, scalable training jobs, inference serving, and deployment monitoring.
- Design and evolve the ML platform for research and engineering teams, covering feature pipelines, training orchestration, deployment workflows, model registries, and experiment tracking tools.
- Design and implement performant data pipelines for large-scale feature extraction and model training data workflows across batch, streaming, and distributed compute environments.
- Optimize ML models and supporting infrastructure for latency, memory footprint, and throughput.
- Collaborate with researchers, data engineers, and infrastructure engineers to improve modeling workflows end-to-end.
Requirements
- Strong software engineering fundamentals with experience building production-grade ML systems.
- Deep ML engineering experience, including training and deploying ML models in production and understanding the full model lifecycle.
- Experience building or maintaining ML infrastructure (e.g., experiment tracking, feature stores, training pipelines, model serving frameworks, or distributed training orchestration).
- Experience building high-throughput, reliable data pipelines.
- Proficiency in Python and PyTorch.
- Strong foundations in linear algebra, optimization, and statistics.
- Awareness of system performance and profiling for efficiency through parallelization, data loading, or model optimization.
- Comfort reading ML papers, questioning modeling choices, and proposing improvements while considering engineering realities.
Skills
- Python
- PyTorch
- JAX
- TensorFlow
- CUDA
- C++
- Spark
- Ray
- Linear Algebra
- Optimization
- Statistics
- MLOps
- Data Engineering
- Deep Learning
- Neural Networks
- Sequence Models
Location
- New York City
Work Type
- Onsite (4 days per week)
Experience Level
- ML Engineering
- Data Engineering
- ML Platform/MLOps
Benefits
- Health insurance with 100% premium covered and access to 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
- Engaged and active community with company events, affinity networks, and extracurricular interest groups
About the Company
- AIA is Bridgewater Associates' dedicated Artificial Intelligence research and investment lab.
- 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.
- 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.
Equal Opportunity
- Bridgewater Associates, LP is an Equal Opportunity Employer
