Staff ML Engineer

3 weeks ago

Visakhapatnam, India Yuna.io Full time

Company Description

Yuna's mission is to radically transform the way mental health support is accessed and delivered. By providing 24/7 instant and private support at the push of a button, we aim to address the significant gaps in mental health care accessibility. In a system where waiting times for therapists can reach up to 48 days and many people can't even afford care, Yuna is the answer to immediate mental health support needs.

We’re growing quickly, and every person who joins us plays a pivotal role in shaping the future of mental health technology.


Position Overview

You’ll own Yuna’s core conversational intelligence (i.e. spanning retrieval, memory, safety, evaluation, and post-training). You’ll ship user-visible improvements quickly while raising the bar on empathy, reliability, and guardrails. This is a hands-on staff-level role: equal parts research-to-production and platform thinking.


What you’ll do

  • Lead the end-to-end model lifecycle of Yuna’s conversational AI: problem framing → data curation → modeling/fine-tuning/post-training → evals → rollout → monitoring
  • Work closely with clinical psychologists (both internal and external to Yuna), ensuring the efficacy of Yuna’s conversational AI
  • Design agentic flows (tool use, planning, multi-turn memory) and safe retrieval for mental-health use cases
  • Build and own the evaluation flywheel: benchmarks, judge models, rubrics for conversational quality, ability to help, safety
  • Implement alignment & safety layers (e.g., self-harm detection, red-teaming, refusal strategies, escalation paths) tuned to clinical guidelines
  • Optimize for latency, cost, and quality
  • Partner tightly with product, sales, and users to scope new features and
  • Establish LLMOps: datasets, prompts/checkpoints/versioning, observability, drift monitoring, and incident response for model quality


What you’ll bring

  • 5+ years in ML/AI with 3+ shipping LLM/NLP systems in production
  • Deep skill in Python and modern deep learning stacks (PyTorch/JAX/TensorFlow), retrieval/RAG, embeddings, reranking, and long-context summarization
  • Experience building agentic systems (tool calling, decomposition, memory) and rigorous eval harnesses (offline + online)
  • Strong product sense; you care about user outcomes, not just model metrics
  • Comfort working with ambiguity and creating systems 0 to 1


Location: Remote (work from anywhere, with at least 4 hours overlap with PST)

Employment Type: Full Time


What We Offer

  • Competitive salary (based on experience) + equity options
  • Remote-first culture with flexibility
  • A fast-growing, talented, and empathetic team dedicated to transforming mental health care
  • The opportunity to make a real impact on the world, building cutting-edge AI systems that improve lives every day

  • AI & ML Engineer

    2 weeks ago

    Greater Visakhapatnam Area, India Aspire Spectrum Full time ₹ 12,00,000 - ₹ 36,00,000 per year

    AI & ML EngineerLocation:Visakhapatnam (On-site)Experience:2+ yearsAbout the RoleAspire Spectrum is looking for askilled and passionate AI & ML Engineerto join our growing team. In this role, you will be responsible for theend-to-end lifecycle of AI/ML projects— from research, design, and development to deployment and maintenance. You will work on...

  • Software Engineer

    2 weeks ago

    Visakhapatnam, India Brandsmashers Tech Full time

    Company Description Brandsmashers Tech is a trusted technology partner providing a full spectrum of IT services, including IT Staff Augmentation, Product Development, Web & E-Commerce Solutions, Custom Software Development, UI/UX Design, and AI/ML Solutions. We empower businesses to build, scale, and innovate with confidence by offering expertise, agility,...

  • AI Engineer

    5 days ago

    Visakhapatnam, India TalentBridge Full time

    Job Title: AI EngineerJob Type: 6-Month Contract, after 6 months it will convert to fulltimeJob DescriptionWe are looking for an experienced AIML Engineer with 4–8 years of expertise in AI/ML solutions, specifically in building intelligent applications leveraging LLM orchestration tools.The ideal candidate will have hands-on experience with Semantic...

  • AI Engineer

    5 days ago

    Visakhapatnam, India TalentBridge Full time

    Job Title: AI EngineerJob Type: 6-Month Contract, after 6 months it will convert to fulltimeJob DescriptionWe are looking for an experienced AIML Engineer with 4–8 years of expertise in AI/ML solutions, specifically in building intelligent applications leveraging LLM orchestration tools.The ideal candidate will have hands-on experience with Semantic...

  • Founding engineer

    3 weeks ago

    Visakhapatnam, India Jewelbyte Full time

    We're building the cursor for jewelry production.At Jewelbyte, we’re tackling a frontier problem: training AI systems to design and produce jewelry CAD files with the skill and creativity of expert human designers. This means combining AI, reinforcement learning, and 3 D geometry to create production-ready jewelry designs at scale. It’s a bold challenge...

  • Machine learning engineer

    2 days ago

    Visakhapatnam, India GoML Full time

    Looking for a culture to thrive & build a rewarding career for yourself, join the core team of young hustlers building the next generation of Machine Learning platform & services. You will develop training and deployment pipelines for machine learning, implement model compression algorithms, and productionize machine learning research solving challenging...

  • Miracle Software Systems

    1 week ago

    Visakhapatnam, India Miracle Software Systems Full time

    Position : AI Engineer - Senior AssociateDuration : Full TimeExp : 5+Location : vishakhapatnamPrimary skills : AI,ML,GCP,PYTHONJob Description : We are looking for an AI/ML Engineer responsible for designing and implementing AI/ML solutions to address complex business needs through techniques such as deep learning, generative AI, NLP, image processing, and...

  • Artificial Intelligence Engineer

    2 weeks ago

    Visakhapatnam, India Miracle Software Systems Full time

    Job Description : We are looking for an AI/ML Engineer responsible for designing and implementing AI/ML solutions to address complex business needs through techniques such as deep learning, generative AI, NLP, image processing, and automation. The role involves developing algorithms and intelligent applications, conducting large-scale experimentation,...

  • ML Engineer

    3 weeks ago

    Visakhapatnam, India Miracle Software Systems Full time

    The desired candidate should have experience in developing cutting-edge machine learning models and conducting experiments to drive data-driven decisions Proven experience in developing and implementing machine learning models and algorithms Should have proficiency in designing and building scalable machine learning systems that can handle large volumes of...

  • Senior Mechanical Engineer

    3 weeks ago

    Visakhapatnam, India Morrison Hershfield Full time

    Morrison Hershfield is a multi-disciplinary consulting engineering and management firm located throughout the US and Canada. We are a dynamic, employee-owned, team-oriented firm with a diverse project list that ensures you will be working on exciting, high-profile and challenging projects. For more than 75 years, the people at Morrison Hershfield have been...