Lead AI Data Trainer & AI Model Optimization EngineerVerified as a real position

Who we’re looking for

• Experience working with NLP, document intelligence, information retrieval, or machine learning systems.
• Experience developing, training, evaluating, or optimizing machine learning models, including classification, extraction, retrieval, or LLM-based workflows.
• Strong Python development skills and experience working with machine learning frameworks, data processing pipelines, and model evaluation approaches.
• Experience with model training, data annotation, dataset management, MLOps practices, or AI lifecycle automation.
• Understanding of modern AI technologies such as transformer models, embedding models, SLMs, LLMs, RAG architectures, or prompt engineering techniques.
• Experience working with data quality, testing, validation, or model performance measurement frameworks.
• Strong analytical, communication, and problem-solving skills, with the ability to collaborate across technical and business teams.

What you’ll do

• Lead the development, training, evaluation, and optimization of AI models supporting document intelligence, entity extraction, document classification, information retrieval, and compliance automation use cases.
• Define and evolve training pipelines, datasets, evaluation frameworks, and automated workflows that support scalable model development and continuous improvement.
• Guide the optimization of SLMs, FastText classifiers, LLM-assisted workflows, and hybrid AI extraction solutions to meet customer, business, and operational requirements.
• Establish domain intelligence frameworks and best practices that support regulated industries and compliance standards, including GDPR, PCI-DSS, HIPAA, SOC 2, and related requirements.
• Define model quality metrics, validation approaches, and performance standards to improve accuracy, explainability, and operational effectiveness.
• Analyze production behavior, investigate model performance challenges, and collaborate with Product, Engineering, and Customer Success teams to drive continuous improvement.
• Mentor team members, share technical expertise, and contribute to AI engineering standards, documentation, and explainability frameworks.

About the position

We’re in an unbelievably exciting area of tech and are fundamentally reshaping the data storage industry. Here, you lead with innovative thinking, grow along with us, and join the smartest team in the industry. This type of work—work that changes the world—is what the tech industry was founded on. So, if you're ready to seize the endless opportunities and leave your mark, come join us.

Benefits

• Innovation : We celebrate those who think critically, like a challenge, and aspire to be trailblazers.
• Growth : We give you the space and support to grow along with us and to contribute to something meaningful. We have been named Fortune's Best Workplaces in Technology™, Fortune's Best Workplaces in the Bay Area™, and certified as a Great Place to Work®!
• Team : We build each other up and set aside ego for the greater good.
• And because we understand the value of bringing your full and best self to work, we offer a variety of perks to manage a healthy balance, including flexible time off, wellness resources, and company-sponsored team events. Check out purebenefits.com for more information.