We're looking for an engineer who is equally at home running a GPU training platform and the large-scale model training that runs on it. You'll be embedded with a healthcare and life-sciences AI client that trains foundation models on a large GKE-based H200 and GB200 cluster. The client is reorganizing so that its foundation-model platform and its agent platform teams work more closely together.
The role centres on the training stack and the scientists who use it. Solid infrastructure knowledge and experience are required, but the client already has strong SRE and infrastructure capacity, so running the cluster isn't the focus: you'll partner with that team and understand the training workloads well enough to tune, benchmark, and debug them end to end. You'll work alongside other Egen engineers and client scientists, and you'll own your area outright.
What You Will Do:
- Training platform
- Work with the client's infrastructure team on the GKE training platform (GPU node pools, job queueing and scheduling, quota and capacity) from the training workload's side.
- Diagnose and fix platform failures at the root (for example a controller that keeps running out of memory) rather than restarting them.
- Tune multi-node GPU networking (RDMA/GPUDirect, NCCL) for training throughput. Distributed training
- Run, scale and debug multi-node distributed training with PyTorch (FSDP, DeepSpeed, Megatron or NeMo-style) and Ray Train.
- Profile jobs and remove bottlenecks across compute, memory, input pipelines and interconnect. Benchmarking and evals
- Design and run performance benchmarks for the cluster and training stack: throughput, scaling efficiency, cost per run.
- Build and maintain model evaluation harnesses so changes to the platform or training recipes are measured, not assumed.
- Instrument training and the platform with telemetry and monitoring, and keep end-to-end pipeline tests running against what users actually run.
- Treat the platform as your own: proactively find issues, plug gaps and propose solutions before anyone asks, keep the team informed of status and risks, and look after the scientists who use it.
- Explain and defend your design decisions to client engineers and scientists in detail.
- Stay effective while the client's priorities and team structure change.
Your Technical Toolkit:
- Core Languages: Mastery of Python and shell scripting
AI/LLM Ecosystem: Extensive experience with Google Gemini, GPT-4, or LLaMA; deep knowledge of Prompt Engineering and Fine-tuning
Data & Search: Expertise in Vector Databases (Vertex AI Vector Search, pgvector, etc.) and implementing Semantic Search
Infrastructure: Hands-on experience with Google Cloud (Vertex AI) and building scalable software architectures
Frameworks: Proficiency in LangChain, LlamaIndex, or similar orchestration layers
Mindset: A strong software engineering foundation—you write clean, maintainable code and understand the full SDLC
Basic Qualifications:
5+ years of engineering experience, with recent hands-on work on GPU training infrastructure and large-scale training.
Strong Python software engineering: production-quality code, tests and code review. Coding is central to this role, not an afterthought.
GKE and GPU scheduling: Kubernetes on Google Cloud with GPU node pools, job queueing (Kueue or similar), and capacity and quota management.
Distributed PyTorch and Ray Train: has personally run and debugged multi-node training jobs.
RDMA and GPU networking: GPUDirect/RDMA, NCCL tuning, multi-node GB200 (or H200) topology.
Benchmarking and evals: has designed performance benchmarks and model evaluations, and used them to make decisions.
Deep enough understanding to explain how and why a system works, without leaning on AI coding tools to carry the reasoning.
A track record of visible ownership: present, communicative, and accountable on a demanding team.
Personal Attributes:
Healthcare or life-sciences data (genomics, clinical, biomedical), and working under HIPAA.
Observability and telemetry for training platforms: metrics, tracing, alerting, end-to-end pipeline testing.
Similar Jobs
What you need to know about the Boston Tech Scene
Key Facts About Boston Tech
- Number of Tech Workers: 269,000; 9.4% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Thermo Fisher Scientific, Toast, Klaviyo, HubSpot, DraftKings
- Key Industries: Artificial intelligence, biotechnology, robotics, software, aerospace
- Funding Landscape: $15.7 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Summit Partners, Volition Capital, Bain Capital Ventures, MassVentures, Highland Capital Partners
- Research Centers and Universities: MIT, Harvard University, Boston College, Tufts University, Boston University, Northeastern University, Smithsonian Astrophysical Observatory, National Bureau of Economic Research, Broad Institute, Lowell Center for Space Science & Technology, National Emerging Infectious Diseases Laboratories
