Member of Technical Staff, Search
Core
Design, train, and improve state-of-the-art semantic search models and data pipelines to enhance information retrieval accuracy and efficiency.
Role type
Senior IC machine-learning engineer (search/retrieval)
Builds
Intelligent, efficient, and precise search systems (embedding and reranker models)
Domain
AI / Machine Learning / Information Retrieval
Deliverable
production ML models
Required skills
Python, PyTorch, TensorFlow, TF-Serving, JAX, XLA/MLIR, Large Language Models, information retrieval models, distributed training strategies, GPU optimization
Preferred skills
C++, Golang, building training/evaluation datasets
Technologies
PyTorch, TensorFlow, TF-Serving, JAX, XLA/MLIR, GPUs
Responsibilities
Design, train, and improve cutting-edge search models; Gather high-quality retrieval datasets and optimize data pipelines; Work with model serving team to ensure fast and stable inference; Collaborate with product teams to develop solutions; Engage in research collaborations and publish work in top-tier conferences
Seniority
Senior, hands-on IC