Member of Technical Staff - Environments (ML)

Remote, Full-Time

As an Environment Engineer (ML), you will build on top of our core platform to create the simulation environments in which frontier coding agents learn to be proficient in Machine Learning. You will build testing infrastructure on top of large codebases to grade the solutions of coding agents on real world ML engineering tasks and build automations to speed up parts of testing and task creation.

Here are some things you might work on:

• Write an end-to-end test for an RL framework that trains a small model to ensure that the reward of our model is going up as expected or numerical instabilities occur
• Find research papers proposing algorithmic improvements for model training, implement them and write extensive tests to ensure that we can test whether models are able to reimplement them

This role will require you to constantly work with new codebases; hence, you will need to be able to quickly get familiar with new technologies. The ideal candidate has multiple years of experience working with ML frameworks like PyTorch or Jax and has trained large models on large datasets using large amounts of GPUs.

Interview Process

Our interview process is designed to be quick. We usually have 2-3 technical interviews before making an offer. You can expect to have an offer within seven days in most cases.

Apply

Before Proximal, our engineering team built critical AI infrastructure at companies like Cursor, OpenAI, Prime Intellect and Browserbase. We are growing extremely fast and are backed by top-tier funds and angel investors from OpenAI, Anthropic, xAI and Thinking Machines.

We're excited to work with smart and creative team members - if you would like to join us, please send your resume to hiring@proximal.so. In your email, please include two to three sentences about your prior experience and tell us what your favorite developer tool is (except Cursor).