Machine Learning Engineer

Company Overview

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 right training 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 improves 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 as 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, PearVC, and the founders of Dropbox and Gradient Ventures.

Our team comes with over a decade of industry and research experience from top organizations and labs like CMU, Berkeley AI Research Lab, Google Brain, Scale AI, and Qualcomm. We've previously founded and sold 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, ICML, AAAI.

Role Description

As a Machine Learning Engineer at Advex, you will play a pivotal role in shaping the company's technical direction. As an MLE you are well versed with the ML life-cycle process, collect data, label data, train model and iterate multiple times before achieving desired performance. Your hands-on involvement will include working with our CTO to automate the entire ML life-cycle by scaling the Advex pipeline that automatically (1) identifies gaps in data distributions, (2) generates labeled synthetic samples, (3) and trains downstream models on enhanced synthetic datasets. You are also expected to 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.

Sneak peek of things we're working on and what you'll have the opportunity to spearhead:

Research:

1. Understanding gaps in data distributions

2. Controllability of Diffusion Models

3. Evaluation of generative models

Engineering:

1. Full-stack development

2. Large scale model training

3. Building infrastructure to run complex ML pipelines

4. Pruning and model quantization

We are seeking an individual who is strategic, capable of transforming research constraints and product specifications 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

We are seeking a candidate who embodies the following qualifications and characteristics:

• 3 to 5 years of industry experience in Deep Learning and Computer Vision

• Prior experience working at a startup

• Experiences in end to end ML application development, including data engineering, model tuning, and model serving

• Technical expertise demonstrated through published papers or industry experience:

  • Representation learning – SSL, VAE, Dino, SAM

  • Building and scaling ML Infrastructure - AWS, GCP, containerization

• Proven track record of rapidly delivering complex project within tight deadlines

• Ability to make critical decisions with little information

• Excel at identifying bottlenecks and gaps within research papers and swiftly translating them into code.

Our Values

• Extreme Ownership – Own your piece to an unreasonable level – if that piece succeeds or fails, for whatever reason, the buck stops with you.

• Fierce Velocity – Rome was not built in a day, but they were laying bricks every hour.

• The Best Part is No Part – Simplicity over complexity.

• Be Epsilon Greedy – Greatness cannot always be planned; be willing to explore novel ideas even if things don’t work out.

• Customer Obsessed – Our customers are the backbone of humanity. It is our duty to improve and make them better in order to exponentially improve the lives of billions of people.

• Feedback is the breakfast of champions – Embrace feedback. A healthy team is one where constructive feedback is welcomed and expected.

• Argmax (team) – The only way to win is together; One Team. One dream

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.