Back to jobs

Head of ML Infra

Impax Recruitment
San Francisco Bay Area
Full-time
30,000,000 – 50,000,000 / year

We're partnered with a Series A startup, who are building large scale foundation models which build relationships between visual and textual data to predict and control the future of the real world.

They're building one of the most talent dense startup teams in the Bay, with engineers from top AI labs and foundation model startups.

If you have experience with at least one of the below, please apply:

* Large scale distributed training, with 100's if not thousands of GPU's.

* Large scale inference, for multibillion parameter models.

* Model serving/training platform development.

* Open source contributions to relevant frameworks such as Deepspeed, vLLM, Ray etc...

What we can offer:

* Relocation from anywhere in the world if the criteria is met.

* IC work, with a technical leadership role to work directly with the founders to shape the infra direction.

* $300k - $500k base + significant equity.

* No red tape/bureaucracy - get things done mentality.

Applications go to the hiring team directly