AI/ML Developer

2 - 3 years

0 Lacs

Posted:1 day ago| Platform: Linkedin logo

Apply

Work Mode

On-site

Job Type

Full Time

Job Description

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.


  • Mock Interview

    Practice Video Interview with JobPe AI

    Start Python Interview
    cta

    Start Your Job Search Today

    Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

    Job Application AI Bot

    Job Application AI Bot

    Apply to 20+ Portals in one click

    Download Now

    Download the Mobile App

    Instantly access job listings, apply easily, and track applications.

    coding practice

    Enhance Your Python Skills

    Practice Python coding challenges to boost your skills

    Start Practicing Python Now

    RecommendedJobs for You