Machine Learning Engineer Full-time

Who We Are:

AI Fund is a venture studio that works with entrepreneurs to build companies rapidly and increase the odds of success. Backed with $300m from Greylock, NEA, Sequoia and other LPs, and spearheaded by Andrew Ng (Co-Founder of Coursera, Co-Founder/Head of Google Brain), we’re partnering with our CEOs to build the next generation of AI tech companies.

What You Will be Doing:

We are currently seeking an exceptionally talented Machine Learning Engineer to join our technical team. You will be responsible for the end-to-end ownership of scalable machine learning systems-from translating business ideas into actionable technical solutions, to building data pipelines, training models, and analyzing performance in a production environment. You will be joining a team that leads the development from Proofs of Concept (PoCs) to initial production pipelines in early-stage startups. You must be adaptable, able to assume multiple roles, and thrive in a dynamic environment.

Main Responsibilities:

• Design and Develop Production-Ready Solutions: Lead the development of new machine learning models focused on transforming a wide variety of datasets and use cases into actionable insights. Apply fundamental machine learning concepts to iterate quickly, debug model-related issues, and develop innovative techniques to handle unique cases for each customer.
• Collect, Process, and Analyze Data: Understand and analyze data as a critical component of our machine learning projects. Build pipelines and processes for data cleaning and organization, and develop tools to facilitate data analysis. Identify areas where models are underperforming and propose effective solutions based on your analysis.
• Stay Current with Advances in Machine Learning and Generative AI: Keep up-to-date with the state of the art in machine learning and generative AI. Continuously integrate relevant advancements into our models and workflows to maintain a cutting-edge technological edge.

What You Must Bring:

• Experience: Minimum of 3 years of professional experience in designing, training, and deploying machine learning models.
• Technical Expertise:
• Strong foundation in computer science, including data structures, algorithms, and design patterns.
• Expertise in Python, demonstrated by implementing multiple medium to large-scale projects.
• Proven ability to implement and debug machine learning models.
• Familiarity with machine learning frameworks and libraries (e.g., scikit-learn, PyTorch, MLFlow).
• Experience in setting up data pipelines.
• Experience in developing RESTful APIs.
• Proficiency in cloud platforms (AWS, GCP) and containerization technologies.
• Industry experience with relational databases, SQL, and NoSQL tools.
• Experience with commercial and open-source Large Language Models (LLMs) and frameworks such as LangChain and LlamaIndex.
• Education: Bachelor’s degree in Computer Science, Mathematics, or a related field.
• Nice to Haves:
• Experience with agentic frameworks such as LangGraph and crewAI.
• Experience designing and deploying retrieval augmented generation (RAG) systems into production.
• Research experience in machine learning or artificial intelligence.
• Contributions to open-source machine learning projects.
• Experience with Agile development methodologies.
• Master’s or Ph.D. degree in a relevant field is a plus.
• Soft Skills:
• Excellent communication skills with the ability to engage in in-depth technical discussions with both engineering teams and business stakeholders.
• Proficiency in English.
• Self-starter with the ability to work comfortably in an early-stage environment.

AIF’s Core Values

• Teamwork: We work together to achieve goals as a team. We care for and watch out for each other. We share responsibilities as needed, even when it falls outside of our day-to-day functions.
• Trust: We trust each other to do their job to the best of their ability and assume positive intent. We value an environment where mistakes are viewed as opportunities to learn and improve.
• Transparency: We actively listen with humility and respect, understand different perspectives, and care about each other’s growth. We are candid, honest, and constructive in our communications with anyone we encounter.
• Execution-Oriented: We identify the highest value work to be done and then move efficiently and diligently to complete that work. Our attitudes and behaviors reflect our unwavering commitment to both productivity and excellence.
• Integrity: We are responsible and truthful. We uphold high moral standards in everything we do. We will not compromise our principles for short-term rewards.

At AI Fund, we are committed to providing an environment of mutual respect where equal employment opportunities are available to all applicants without regard to race, color, religion, sex, pregnancy (including childbirth, lactation and related medical conditions), national origin, age, physical and mental disability, marital status, sexual orientation, gender identity, gender expression, genetic information (including characteristics and testing), military and veteran status, and any other characteristic protected by applicable law. AI Fund believes that diversity and inclusion among our employees is critical to our success as a company, and we seek to recruit, develop and retain the most talented people from a diverse candidate pool. Selection for employment is decided on the basis of qualifications, merit, and business need.