Machine Learning Systems Architect

Company Description

VIWELL is a wellness-focused digital platform designed to improve employee well-being, engagement, and productivity while empowering business leaders with actionable insights. With a robust dashboard, VIWELL delivers real-time analytics to craft strategic well-being initiatives. Through an intuitive employee app, the platform enables engagement with events, rewards, gamification, and personalized wellness plans, complemented by resources like the Employee Assistance Program (EAP). By addressing the health needs of both employers and employees, VIWELL bridges the gap between individual well-being and organizational success.

Role Overview

The Data & Machine Learning (ML) Engineer will play a key role in enabling VIWELL’s next-generation personalization and intelligent recommendation systems. The ideal candidate brings strong experience in data engineering, applied machine learning, and safe AI practices , preferably within digital product environments and fast-paced or startup settings.

This role requires a mix of hands-on engineering capability, analytical depth, and safety-driven mindset. The Data & ML Engineer will work closely with Product, Backend Engineering, Analytics, and AI Leadership to deliver high-quality events, features, eligibility rules, ranking logic, and safe GPT rewrites that enhance user engagement and personalization across VIWELL’s ecosystem.

The role directly supports VIWELL’s mission of scaling to millions of users through intelligent, privacy-preserving, and high-performance data and ML systems.

Key Responsibilities

• Define event schemas and data contracts in collaboration with Product and Analytics to ensure high-quality tracking and feature availability.
• Engineer and maintain data features in Databricks with robust quality checks, lineage documentation, and automated orchestration.
• Design and implement eligibility rules and ranking logic to power deterministic, safe personalization and recommendations.
• Build guardrails and validation systems including blocklists, allowlists, rewrite validators, and pseudonymization frameworks for safe GPT-assisted rewrites.
• Design A / B experiments , exposure logging, and statistical frameworks with power analysis to support iterative product improvements.
• Integrate ML and personalization systems with backend APIs, ensuring optimized response times below 300 milliseconds.
• Collaborate with cross-functional teams to ensure clarity, alignment, and seamless execution of data and ML solutions.
• Stay informed on evolving AI safety principles, data quality standards, and best practices in ML engineering.
• Continuously improve feature pipelines, data freshness, ranking performance, and validation accuracy.

Qualifications & Experience

Key Competencies

What We Offer

Location : India (Hybrid) — applicants must be based in India.

How to Apply :

If you meet the criteria and are ready to be part of a purpose-driven company, send your CV to : career@viwell.com

Join us and help transform well-being across the region!

#VIWELL #ML #IndiaJobs #UAE #HybridWork #SaaS #Wellbeing