Machine Learning Engineer

Engramme is hiring a Machine Learning Engineer in San Francisco to build and run production-grade ML systems for memory retrieval and contextual understanding.

Responsibilities

• Design, implement, and iterate on ML models for memory retrieval and ranking
• Develop ML infrastructure and training pipelines that support scalable experimentation
• Deploy models into production and monitor performance at scale
• Optimize model performance and latency for real-time use cases
• Work with vector databases and embedding systems
• Implement MLOps best practices, including monitoring and operational workflows
• Partner with the research team to productionize new algorithms and improve reliability

Requirements

• 4+ years of ML engineering experience
• Strong proficiency in Python and ML frameworks including PyTorch and TensorFlow
• Experience deploying ML models to production
• Knowledge of MLOps, model serving, and monitoring
• Understanding of NLP, embeddings, and retrieval systems
• Experience with AWS and/or GCP and containers
• Strong software engineering fundamentals
• Experience with large-scale data processing

Technologies

• Python
• PyTorch
• TensorFlow
• AWS
• GCP
• Containers
• Vector databases
• Embedding systems

Nice to Have

• Experience with large language models and prompt engineering
• Knowledge of vector databases such as Pinecone, Weaviate, or Milvus
• Experience with recommendation systems or search
• Background in information retrieval or ranking systems
• Knowledge of model optimization and quantization
• Experience with real-time ML systems
• Familiarity with transformer architectures