AI/ML Developer

Location: Bangalore 2- 3 years Experience preferred.

Company:

About the Role

WYH is building intelligent health solutions that power health assessments and exercise evaluations. We are looking for an AI/ML Developer with 2–3 years of experience who is excited to work on real-world healthcare problems, experiment rapidly, and contribute to the AI models powering our SaMD assessment platform.

In this role, you will:

• Work directly with the product and engineering leads on core AI features.
• 
  
• Apply practical AI/ML techniques to improve detection, tracking, and assessment algorithms.
• 
  
• Help build data pipelines, experiment frameworks, and deploy ML models at scale.
• 
  
• Contribute to the future of AI-driven physiotherapy and digital health.

Key Responsibilities

1. Model Development & Experimentation

• Develop and refine ML models for face, tongue, and movement-based exercise recognition.
• 
  
• Work with CV/NLP/LLM components as required—classification, detection, tracking, embeddings, or lightweight reasoning systems.
• 
  
• Evaluate models using real-world physiotherapy data and improve performance across devices (including low-end Android).

2. Data Engineering & Pipeline Management

• Clean, preprocess, and structure datasets for training and evaluation.
• 
  
• Work with image/video datasets, sensor data, and annotation workflows.
• 
  
• Build automated scripts for dataset generation, augmentation, and validation.

3. AI Agent & Workflow Integrations

• Contribute to modular AI agents powering guidance, feedback, or conversational layers.
• 
  
• Basic understanding of RAG, embeddings, tool usage, or agent frameworks.

4. Deployment & Performance Optimization

• Package and deploy ML models (on-device or cloud).
• 
  
• Optimize model performance—latency, size, inference speed.
• 
  
• Work with engineering team to integrate models into Flutter/Android/iOS applications.

5. Backend Development

• Build and maintain backend APIs that power ML inference, assessment workflows, and reporting pipelines.
• 
  
• Implement model-serving endpoints using Python (FastAPI preferred).
• 
  
• Work with Firebase/GCP services for data storage, real-time syncing, and function triggers.
• 
  
• Collaborate with mobile teams to integrate backend and model outputs into the app.
• 
  
• Ensure backend reliability, performance, and scalability.

Required Qualifications

• Bachelor’s degree in Computer Science, Engineering, AI/ML, or related field.
• 
  

• 2–3 years of hands-on experience in ML/AI development.

• 
  
• Solid understanding of ML fundamentals (classification, regression, CV basics, evaluation metrics).
• 
  
• Practical experience with

  Python

  ,

  PyTorch or TensorFlow

  ,

  OpenCV

  ,

  NumPy

  ,

  Pandas

  .
• 
  
• Familiarity with training ML models on real datasets.
• 
  
• Knowledge of deploying models using APIs or lightweight runtime environments.
• 
  
• Basic understanding of cloud platforms (preferably

  GCP

  , optionally AWS/Azure).
• 
  
• Experience with version control (Git) and collaborative development.

Good to Have (Bonus Skills)

• Experience with

  on-device ML

  , TFLite, CoreML, or model quantization.
• 
  
• Exposure to

  LLMs, embeddings, RAG

  , or conversational AI agents.
• 
  
• Knowledge of physiotherapy, healthcare AI, or biometric systems.
• 
  
• Familiarity with MLOps tools like Docker, FastAPI, model registries, or monitoring tools.
• 
  
• Experience working in a startup or fast-paced product environment.

Why Join Watch Your Health?

• Work on meaningful tech impacting healthcare outcomes.
• 
  
• Build state-of-the-art AI models for real-world health assessment applications.
• 
  
• Ownership, autonomy, and direct impact from day one.
• 
  
• Opportunity to grow with a fast-scaling healthtech startup.