Machine Learning Engineer

Job OverviewTake the next step in your career now, scroll down to read the full role description and make your application.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 UsAdvex 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 DescriptionAs 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:ResearchUnderstanding gaps in data distributionsControllability of Diffusion ModelsEvaluation of generative modelsEngineeringFull‐stack developmentLarge scale model trainingBuilding infrastructure to run complex ML pipelinesPruning and model quantizationWe are seeking an individual who is strategic, capable of transforming research constraints and product specifications into an enterprise‐ready application. We value self‐stators who aren't afraid to dive into the details and iterate on prototypes in the early stages of development.About YouWe are seeking a candidate who embodies the following qualifications and characteristics:3 to 5 years of industry experience in full‐stack Deep Learning and Computer VisionPrior experience working at a startupExperiences in end‐to‐end ML application development, including data engineering, model tuning, and model servingTechnical expertise demonstrated through published papers or industry experience:Representation learning – SSL, VAE, Dino, SAMBuilding and scaling ML Infrastructure - AWS, GCP, containerizationProven track record of rapidly delivering complex project within tight deadlinesAbility to make critical decisions with little informationExcel at identifying bottlenecks and gaps within research papers and swiftly translating them into code.Our ValuesExtreme 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 dreamNote: we are a company founded by immigrants, and we are committed to providing support to immigrant workers throughout their journey. xevrcyc This includes offering assistance to international students and workers with various types of visas, such as OPT, H1, EB, and more.