ML Engineer at Proxima | Boston, MA, US | Rezi

ML Engineer at Proxima

ML Engineer

Proxima · Boston, MA, US

4 weeks ago

ML Engineer

Proxima · Boston, MA, US

a month ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this ML Engineer role.

Rezi rewrites your resume against Proxima's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the ML Engineer posting at Proxima — free, in seconds.

About the Role

Proxima is advancing an AI-native approach to drug discovery by making protein interactions programmable. We are looking for a versatile software engineer to build the tools and systems behind our drug discovery research, working across automation, infrastructure, data, and ML to help scientists develop models, run experiments, and use the results.

Responsibilities

  • Build reliable scientific workflows that colleagues can run, reproduce, and troubleshoot with less manual intervention.
  • Integrate new models, datasets, and evaluation methods while preserving data correctness and compatibility with existing experiments.
  • Improve compute workflows through better job submission, monitoring, failure recovery, and artifact tracking.
  • Identify and remove bottlenecks in data preparation, execution, and evaluation so researchers can iterate faster.
  • Deliver libraries, services, and developer tools that are easy to use and maintain, with appropriate tests, documentation, and monitoring.
  • Contribute to the culture of a rapidly growing company.
  • Be challenged by colleagues and learn something new every day.

Requirements

  • Enjoy taking a problem from investigation through deployment.
  • Be able to integrate a dataset into training, diagnose a stalled inference job, build an API for scientific results, or automate a manual workflow.
  • Be able to find your way around an unfamiliar codebase, understand the constraints, and deliver a focused, well-tested improvement.
  • Strong Python and software engineering fundamentals, including data structures, interface design, testing, concurrency, and systematic debugging.
  • A track record of delivering software that other people use and depend on, including maintaining and troubleshooting it after release.
  • Experience building or supporting ML or scientific computing workflows, with an understanding of how data, model execution, and evaluation fit together.
  • Experience running software on Linux, working with containers, and shipping changes through automated tests and CI/CD.
  • Ability to break down ambiguous problems, make progress independently, and explain technical tradeoffs with evidence.
  • Willingness to work across automation, infrastructure, data, and ML, with a strong sense of ownership and clear communication with scientists and engineers.
  • Ability to use AI development tools effectively and take responsibility for the correctness and maintainability of the resulting code.
  • We welcome strong engineers from other technical domains; a biology or chemistry background is not required.

Skills

  • Python
  • PyTorch
  • Containerization
  • Kubernetes
  • Cloud storage
  • GPU compute
  • High-performance cloud computing
  • Systems engineering
  • Machine learning
  • Linux
  • Automated tests
  • CI/CD
  • AI development tools
  • Workflow orchestration
  • Databases
  • Computational biology
  • Computational chemistry

Location

  • Remote

Work Type

  • Full-time

Experience Level

  • Mid-level

About the Company

  • Proxima (formerly VantAI) is advancing an AI-native approach to drug discovery by making protein interactions programmable.
  • Our platform brings together foundation-model machine learning, a scalable data generation engine, and a partnership track record exceeding $5B in collaborations across the world’s leading biopharma and tech organizations.
  • We’ve recently closed an oversubscribed seed round partnering us with an elite group of sophisticated and dedicated VCs including DCVC, Nvidia’s Nventures, AIX, and Yosemite among others.
  • Neo-1 is our all-atom foundation model that combines state-of-the-art structure prediction and molecular generation in a single system.
  • Neo-1 enables rapid exploration of chemical and structural space for high-value, previously intractable targets, and in particular unlocks small molecule proximity therapeutics like molecular glues with AI for the first time.
  • In parallel, we are developing an advanced structural interactomics platform (NeoLink) built on proprietary XLMS technology and a lab equipped with next-generation mass spectrometry instrumentation.
  • This platform produces proteome-scale maps of protein interactions and helps identify small molecules that modulate proximity.
  • Together with Neo-1, it creates an integrated system capable of co-folding protein complexes while generating candidate small molecules to influence those interactions.
  • Proximity-based therapeutics represent one of the most promising frontiers in modern drug discovery with the potential to treat previously intractable diseases and target ‘undruggable’ proteins.
  • Our technology combines proteome-scale structural data with state-of-the-art generative AI foundation models and, coupled with our talented team of scientists and engineers, we are uniquely well-positioned to discover and develop a new class of medicines.
  • Come join us!