AI/ML Engineer Text Chunking & Embedding Benchmarking

AI/ML Engineer - Text Chunking & Embedding Benchmarking 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 :

4078823 Job Summary AI/ML Engineer - Text Chunking & Embedding Benchmarking Remote Required 8+ years in AI/ML Engineer to benchmark and optimize text chunking and embedding strategies for large-scale clinical data (up to 1.5B documents). The role focuses on building scalable pipelines, evaluating models, and delivering recommendations aligned with PHI compliance and high-throughput requirements. Key Responsibilities Evaluate embedding models and chunking strategies on clinical data Benchmark across cost, performance, scalability, and quality Design high-throughput embedding pipelines Deliver comparison reports and recommendations for large-scale deployment Build a reusable evaluation framework for future models Ensure compliance with PHI/HIPAA and governance standards Required Skills & Experience Strong experience in AI/ML, NLP, and embeddings Hands-on with OpenAI, Cohere, SBERT or similar models Experience with large document chunking (>200 pages) Knowledge of distributed systems and cloud (Azure/AWS/GCP) Understanding of healthcare data compliance (PHI/HIPAA) Preferred Experience with large-scale embedding pipelines (billions of records). Prior work on healthcare data platforms. Understanding of audit logging frameworks and lineage tracking. Ability to collaborate with cross-functional teams in regulated environments. 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."