AI/ML Engineer Vector Store & Retrieval Systems

AI/ML Engineer - Vector Store & Retrieval Systems Image

Skill Set :

AI/ML Engineering

Total Experience :

6.00 to 8.00 Years

No of Openings :

1

Job Post Date :

29/05/2026

Job Expiry Date :

31/07/2026

Domain :

IT

Location :

FLORHAM PARK

[United States]

Job Reference No :

4078822 Job Summary AI/ML Engineer - Vector Store & Retrieval Systems Remote 10+ years in AI/ML or Data Engineering (healthcare domain preferred) to design, benchmark, and optimize text embedding and vector search solutions for large-scale clinical data. This role focuses on building high-performance pipelines and evaluating AI models to support intelligent search across patient records. Key Responsibilities Develop and benchmark text chunking and embedding strategies Evaluate embedding models (OpenAI, Cohere, SBERT, etc.) Design and optimize vector search systems Build scalable pipelines for large datasets (up to billions of documents) Measure performance across cost, speed, scalability, and accuracy Support near real-time indexing and high query workloads Ensure compliance with PHI/HIPAA and enterprise security standards Required Skills Strong experience in AI/ML and NLP (embeddings, transformers) Hands-on with LLMs, RAG, and vector databases Proficiency in Python and data frameworks (Spark/Databricks) Experience with cloud platforms (Azure/AWS/GCP) Knowledge of distributed systems and high-throughput pipelines Understanding of healthcare data and PHI compliance The pay range for this role is $100k - $130k per annum including any bonuses or variable pay. Tech Mahindra also offers benefits like medical, vision, dental, life, disability insurance and paid time off (including holidays, parental leave, and sick leave, as required by law). Ask our recruiters for more details on our Benefits package. The exact offer terms will depend on the skill level, educational qualifications, experience, and location of the candidate. Thanks & Regards Tech Mahindra is an Equal Employment Opportunity employer. We promote and support a diverse workforce at all levels of the company. All qualified applicants will receive consideration for employment without regard to race, religion, color, sex, age, national origin or disability. All applicants will be evaluated solely on the basis of their ability, competence, and performance of the essential functions of their positions with or without reasonable accommodations. Reasonable accommodations also are available in the hiring process for applicants with disabilities. Candidates can request a reasonable accommodation by contacting the company ADA Coordinator at ADA_Accomodations@TechMahindra.com."