Machine Learning Research Engineer

At Advex, we're working on solving the hardest problem in all of computer vision – data collection. In order to train a reliable AI model, you need access to the righttraining data. The inability to gather the right data has been the primary obstacle in global adoption of computer vision in mission critical industries like manufacturing and industrial automation.

Our technology is guaranteed to improve model performance by 20-30% for any computer vision task while simultaneously reducing production costs by 10x. Our customers include Fortune 500 companies and some of the fastest growing startups in the world. We invite you to join us at the ground level a s we embark on rapid expansion of our product offerings and customer base.

About Us

Advex is a seed stage tech startup backed by Construct Capital (Solidworks), PearVC (DoorDash), Emerson Collective (Laurene Jobs), Kleiner Perkins (Google), and Arash Ferdowsi (Dropbox Founder).

Our team comes with over a decade of industry and research experience from top organizations like Google Brain, Scale AI, Qualcomm, and Keyence. We've founded multiple companies and created products used over 30 million times. Our team has also published research papers and won research competitions at top conferences like ICLR and AAAI.

Role Description

As a Machine Learning Research Engineer at Advex, you will play a pivotal role in shaping the company's technical direction. Your hands-on involvement will include working with our CTO to (1) spearhead diffusion research to solve OOD generalization problem and (2) build our product and engage directly with our customers. As we expand, you will contribute heavily to the architectural roadmap, laying the foundation for the Advex platform. Your technical leadership will be instrumental in supporting and growing our engineering and product teams.

We are seeking an individual who is strategic, capable of transforming product specifications and front-end components into an enterprise-ready application. We value self-starters who aren't afraid to dive into the details and iterate on prototypes in the early stages of development.

About You

• 2 to 5 years of research experience in Computer Vision and Deep Learning

• PhD or extensive research with relevant publications on:

  • Diffusion research – controllability, latent diffusion models, unconditional image generation

  • Representation learning techniques such as BYOL and MAE

• PyTorch and Python experience

• Ability to make critical decisions with very little information

• Proven track record of rapidly delivering complex projects within tight timeframes

• (Bonus) Prior experience working at a startup

• Exceptional communication skills, including the ability to lead in-depth discussions with clarity and conciseness.

• Proficiency in reading academic papers and implementing innovative ideas promptly. You excel at identifying bottlenecks and gaps within research papers and swiftly translating them into code solutions.

• Publications at top conferences like NeurIPS and ICLR

Note: we are a company founded by immigrants, and we are committed to providing support to immigrant workers throughout their journey. This includes offering assistance to international students and workers with various types of visas, such as OPT, H1, EB, and more.