AI Data Trainer & AI Model Optimization Engineer

Pure Storage · Prague, Czech Republic

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.

THE ROLE

We are seeking an AI Data Trainer & AI Model Optimization Engineer to develop, train, validate, and continuously improve AI models that power intelligent document processing, entity extraction, document classification, and compliance automation solutions.

In this role, you will build and maintain training pipelines, generate and validate training datasets, and optimize Small Language Models (SLMs), FastText classifiers, LLM-assisted workflows, and domain-specific extraction engines across regulated and data-intensive industries, including Financial Services, Healthcare, Real Estate, Insurance, and Government.

This role combines data science, machine learning operations (MLOps), document intelligence, automation engineering, and quality assurance practices to support high levels of accuracy, explainability, and operational performance in enterprise production environments.

WHAT YOU'LL DO

  • Develop, train, and optimize AI models supporting document intelligence, entity extraction, document classification, information retrieval, and compliance automation use cases.
  • Build and maintain training pipelines, datasets, evaluation frameworks, and automated workflows that support scalable model development and continuous improvement.
  • Improve and evaluate SLMs, FastText classifiers, LLM-assisted workflows, and hybrid AI extraction solutions to meet customer and business requirements.
  • Design and maintain domain intelligence frameworks that support regulated industries and compliance standards, including GDPR, PCI-DSS, HIPAA, SOC 2, and related requirements.
  • Establish model quality metrics and validation processes, using performance insights to improve accuracy, explainability, and operational effectiveness.
  • Analyze production behavior, investigate model performance issues, and collaborate with Product, Engineering, and Customer Success teams to implement improvements.
  • Create and maintain documentation covering training methodologies, evaluation approaches, model behavior, and explainability frameworks.
  • We are primarily an in-office environment and therefore, you will be expected to work from the Prague office in compliance with Everpure's policies, unless you are on PTO, or work travel, or other approved leave.

WHAT YOU BRING

  • 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.

NICE TO HAVE

  • Intelligent Document Processing (IDP) platforms.
  • OCR, document understanding, semantic search, or enterprise content management systems.
  • Cloud-based AI platforms such as Azure AI, AWS AI, or Google AI.
  • Explainable AI frameworks and model governance practices.
  • Regulated industries and compliance-focused AI solutions.
  • AI observability, automated testing, vector databases, or production AI operations.

WHAT YOU CAN EXPECT FROM US:

  • 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.

ACCOMMODATIONS AND ACCESSIBILITY:

Candidates with disabilities may request accommodations for all aspects of our hiring process. For more on this, contact us at [email protected] if you’re invited to an interview.

OUR COMMITMENT TO A STRONG AND INCLUSIVE TEAM:

We’re forging a future where everyone finds their rightful place and where every voice matters. Where uniqueness isn’t just accepted but embraced. That’s why we are committed to fostering the growth and development of every person, cultivating a sense of community through our Employee Resource Groups and advocating for inclusive leadership.

Everpure is proud to be an equal opportunity employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender, gender identity, gender expression, transgender status, sexual stereotypes, age, status as a protected veteran, status as an individual with a disability, or any other characteristic legally protected by the laws of the jurisdiction in which you are being considered for hire.

Join us and bring your best.

Bring your bold.

Pure and simple.

Software pay context

Based on 7,956 disclosed Software salaries on RoleSuite, the role pays a median of $157K/year, with most offers between $123K and $196K (10th–90th percentile: $101K–$232K).

See the full Software salary breakdown →
Apply →