Similar Jobs
Founding Machine Learning Engineer
Full-Time - Engineering - SF Bay
About Us
We are on a mission to bridge the gap between enterprise business knowledge and data, democratizing data discovery and curation to prepare organizations for the era of generative AI. Today's data tools are overly complex, poorly integrated, and siloed, forcing AI Practitioners and data scientists alike to spend more time wrestling with tools, relying on tribal knowledge, and navigating data lakes rather than doing meaningful data science work. The current landscape of data tools and processes is heavily manual and needs to catch up with the vast amount of data generated daily. With the advent of Gen AI and multi-modality, this challenge has only grown more complex and broken.
Backed by top VC funds, we are committed to making enterprise data AI-ready faster, more reliably, and with a stronger foundation of factual semantic knowledge. This leads to more accurate models, superior outcomes, and better business results. Our team of seasoned data infrastructure and machine learning experts (from LinkedIn, Visa, Truera, Hive, and Branch) has spent the past two decades building bespoke systems to solve these very challenges.
Join our growing team of ML research and data infrastructure experts. We're committed to empowering AI and data scientists to seamlessly integrate semantic learning with generative AI. Be part of our journey to shape the future of enterprise AI.
About the job
We are looking for a Machine Learning Engineer to join our team who is based in the Bay Area or willing to move. The ideal candidate should have expertise in one or more of the following areas: Knowledge Extraction, Natural Language Understanding, Unsupervised Learning, Information Retrieval, and Fine-tuning LLMs. In this role, you'll play a critical part in developing and training the models, pipelines, and methodologies that power our semantic graph systems. We're looking for someone with a strong background in machine learning, natural language processing, LLMs, and semantic technologies, with a proven track record of tackling complex, large-scale machine learning projects.
What You Will be Doing
Build and/or use best-in-class models to extract knowledge from heterogeneous sources
Develop methods to build and evaluate AI Data Graphs
Fine-tuning LLMs with domain-specific context
Work with data infra engineers to develop the best platform for your needs
Prior Experience
MS degree in CS or equivalent
Startup experience is highly preferred
3+ yrs experience in Machine Learning or Knowledge Extraction
3+ yrs experience working with text
Experience working with and fine-tuning language models such as BERT, LLM, SLMs
Experience with NLP tools such as spacy, openNLP, openNER, GLiNER, etc.
Experience with embedding-based retrieval
Strong background in the fundamentals of machine learning
Deployed and maintained ML, NLP or LLM models in production
Strong data manipulation skills using tools such as numpy and pandas
Great communication skills and a team player
Nice to haves
Familiar with LLM ecosystem and best practises of fine-tuning and prompt-engineering
Experience working on ML and data in the cloud
Experience with GPU optimization
Experience working with docker, k8s
Experience working with ray,vllm
