Senior Machine Learning Engineer

This is a fully remote position, open to applicants in United States.

📋 Description

• Design and construct RAG pipelines, which encompass chunk-store schemas, population from Redshift, and signal-based retrieval.

• Oversee the semantic search infrastructure, including embedding pipelines, chunking, hybrid BM25/vector retrieval, and reranking.

• Develop end-to-end recommendation models, covering feature extraction, training, evaluation, and SageMaker inference.

• Create tool definitions, assemble prompts, and select contexts for AI features backed by Bedrock.

• Compute and provide ML features such as historical aggregates, behavioral profiles, and engagement history.

• Build and sustain batch and incremental Bedrock embedding pipelines that ensure freshness through CDC.

• Establish evaluation frameworks for retrieval metrics, model quality monitoring, and A/B testing.

• Collaborate with data engineers to seamlessly integrate ML models into data pipelines and present results via search and analytics APIs.

⛳️ Requirements

• Minimum of 4 years of experience in ML engineering.

• Proven track record of transitioning models or ML features from experimentation to production.

• Proficient in retrieval systems, embeddings, vector similarity, ANN search, BM25, dense retrieval, chunking, reranking, and vector databases such as PGvector, FAISS, Qdrant, Weaviate, and Pinecone.

• Practical experience with RAG, tool utilization/function calling, prompt engineering, and context assembly.

• Familiarity with Bedrock, OpenAI, Anthropic SDK, or LangChain.

• Strong skills in Python and production ML tools, including PyTorch, scikit-learn, Hugging Face, and XGBoost.

• Understanding of data engineering principles, including SQL or Spark, point-in-time accuracy, and awareness of feature leakage.

• Experience with AWS ML infrastructure, specifically SageMaker, Bedrock, S3, and Lambda.

• Knowledge of workflow orchestration tools like Airflow, Prefect, or Kubeflow.

• Capability to construct dependable scheduled ML pipelines that manage failures autonomously.

🏝️ Benefits

• Flexible remote work opportunities.

• Competitive base salary packages.

• Flexible PTO policy with no predetermined number of days.

• Paid maternity, parental bonding, and medical leave.

• Comprehensive Medical, Dental, Vision, and Prescription coverage; employee-only plans cover 100% of premiums and family coverage costs at 50%.

• Complimentary confidential counseling services through BetterHelp.

• 401(k) plan with a 100% employer match on contributions up to 6%.

• Flexible Spending Accounts (FSA) and Health Savings Accounts (HSA) available.

• Life and AD&D Insurance coverage.

• Up to $60 per month subsidized for fitness, mental health, or other wellness-related expenses.

• $300 home office setup bonus awarded after one year of employment.

• Support for continuing education and professional development.

• Company-sponsored team and offsite events, with full support for attendance.