Machine Learning Infrastructure Engineer

4 days ago


Mountain View, California, United States Contextual AI Full time
Job Overview

As a Machine Learning Infrastructure Engineer at Contextual AI, you will be responsible for designing, building, and maintaining the infrastructure that enables the development, deployment, and scaling of machine learning, data, and service pipelines. You will collaborate closely with data scientists and full-stack developers to shape Contextual AI's machine learning services and systems. Your role will be crucial in ensuring the scalability, reliability, availability, and performance of our RAG 2.0 systems.

Key Responsibilities:
  • Design and build scalable machine learning infrastructure to support the training, evaluation, and deployment of large language models.
  • Instrument, monitor, and optimize the performance and reliability of Contextual AI's services infrastructure.
  • Ensure security best practices are applied to application build pipelines and cloud/SaaS infrastructure.
  • Stay updated on the latest trends and best practices in MLOps, security, and AI to continuously optimize RAG infrastructures.
  • Work closely with stakeholders to understand business needs and translate them into technical solutions.
  • Mentor and guide junior AI engineers and team members, promoting technical excellence and knowledge sharing within the team.
Requirements:
  • Education: Master's degree in Computer Science, Engineering, or a related field (Ph.D. preferred).
  • Experience: 5+ years of experience in building highly available, production-grade containerized distributed ML systems.
  • Technical Expertise: 5+ years of experience with ML infrastructure technologies such as Kubernetes, GCP, AWS, ArgoCD, Terraform, CloudFormation, infrastructure as code, containerization, SLURM automation, Linux fundamentals, GitHub Actions, and CI/CD.
  • Cloud Services: Demonstrated experience in managing cloud-based Kubernetes services.
  • Programming Skills: 5+ years of experience in one programming language such as Python, Java, or Go.
  • Machine Learning Knowledge: Familiarity with key concepts in machine learning, natural language processing, and computer vision.
  • Data Skills: Familiarity with data preprocessing, feature engineering, and data visualization.
  • HPC Experience: Familiarity with large GPU clusters and high-performance computing/networking.
  • Leadership: Experience in mentoring and growing junior engineers into successful leaders.
  • Communication: Excellent communication and collaboration skills, with the ability to work effectively in a fast-paced and dynamic environment.
What We Offer:

Contextual AI offers a competitive salary range of $140,000 - $300,000 + equity + benefits (actual compensation will be determined based on experience, location, and other factors permitted by law). We are an equal opportunity employer and comply with all applicable federal, state, and local fair employment practices laws. All qualified applicants will receive consideration for employment without regard to race, color, religion, national origin, ancestry, sex, sexual orientation, gender, gender expression, gender identity, genetic information or characteristics, physical or mental disability, marital/domestic partner status, age, military/veteran status, medical condition, or any other characteristic protected by law.



  • Mountain View, California, United States KinderKids Learning Center Full time

    Job Description**About KinderKids Learning Center**KinderKids Learning Center is a leading provider of educational services, dedicated to delivering high-quality learning experiences to children. Our team is passionate about creating innovative solutions that enhance the learning process.**Job Summary**We are seeking a highly skilled Machine Learning...


  • Mountain View, California, United States Contextual AI Full time

    Job OverviewContextual AI is seeking a highly skilled Machine Learning Infrastructure Engineer to join our team. As a key member of our infrastructure team, you will be responsible for designing, building, and maintaining the infrastructure that enables the development, deployment, and scaling of machine learning, data, and service pipelines.Key...


  • Mountain View, California, United States Photon Full time

    Join Photon's Team as a Machine Learning Infrastructure EngineerGreetings from Photon, a leading Digital Agency that has emerged as one of the world's largest and fastest-growing companies. We work with 40% of the Fortune 100 on their Digital initiatives and are known for our ability to integrate Strategy Consulting, Creative Design, and Technology at...


  • Mountain View, California, United States Photon Full time

    Machine Learning Infrastructure EngineerGreetings from Photon, a leading Digital Agency!We're seeking a talented Machine Learning Infrastructure Engineer to join our team in Mountain View, CA.About Photon:Photon has emerged as one of the world's largest and fastest-growing Digital Agencies. We work with 40% of the Fortune 100 on their Digital initiatives and...


  • Mountain View, California, United States Photon Full time

    Machine Learning Infrastructure EngineerGreetings from Photon, a leading Digital Agency!We're seeking a talented Machine Learning Infrastructure Engineer to join our team in Mountain View, CA.About Photon:Photon has emerged as one of the world's largest and fastest-growing Digital Agencies. We work with 40% of the Fortune 100 on their Digital initiatives and...


  • Mountain View, California, United States Photon Full time

    Machine Learning Infrastructure EngineerGreetings from Photon, a leading Digital Agency!We're seeking a talented Machine Learning Infrastructure Engineer to join our team in Mountain View, CA.About PhotonPhoton has emerged as one of the world's largest and fastest-growing Digital Agencies. We work with 40% of the Fortune 100 on their Digital initiatives and...


  • Mountain View, California, United States Photon Full time

    About PhotonWe are a leading Digital Agency, working with 40% of the Fortune 100 on their Digital initiatives. Our expertise lies in integrating Strategy Consulting, Creative Design, and Technology at scale.Job OverviewWe are seeking a highly skilled Machine Learning Infrastructure Engineer to join our team in Mountain View, CA.Key ResponsibilitiesDesign and...


  • Mountain View, California, United States TikTok Full time

    About TikTok U.S. Data SecurityTikTok is the leading destination for short-form mobile video. Our mission is to inspire creativity and bring joy. U.S. Data Security ("USDS") is a subsidiary of TikTok in the U.S. This new, security-first division was created to bring heightened focus and governance to our data protection policies and content assurance...


  • Mountain View, California, United States Coupang Full time

    About the RoleCoupang is seeking a highly skilled Senior Staff Machine Learning Infrastructure Engineer to join our Search and Discovery organization. As a key member of our team, you will be responsible for designing and implementing durable and efficient software solutions that handle massive volumes of structured and unstructured data needed to train...


  • Mountain View, California, United States Elite Mente LLC Full time

    Job OpportunityElite Mente LLC is seeking a highly skilled Machine Learning Infrastructure Engineer with GCP to join our team.Location: Remote (with occasional on-site requirements)Duration: 6-12+ months contractInterview: VideoWe are looking for an experienced professional with a strong background in Machine Learning engineering and infrastructure, with a...


  • Mountain View, California, United States Moveworks Full time

    About MoveworksMoveworks is a leading AI company that provides a universal AI copilot for search and automation across all business applications. Our mission is to empower employees to work faster and more efficiently by providing instant knowledge and automating repetitive support issues.Job DescriptionWe are seeking a highly skilled Machine Learning...


  • Mountain View, California, United States Gatik Full time

    About GatikGatik is a leader in autonomous middle mile logistics, delivering goods safely and efficiently using its fleet of light & medium-duty trucks. We focus on short-haul, B2B logistics for Fortune 500 customers, enabling them to optimize their hub-and-spoke supply chain operations, enhance service levels, and reduce labor costs.Job SummaryWe're seeking...


  • Mountain View, California, United States SAMSUNG Full time

    Job SummarySamsung Ads is a leading advertising technology company that leverages machine learning to deliver innovative solutions to advertisers. As a Machine Learning Engineer, you will play a critical role in designing and developing cutting-edge machine learning platforms to support thousands of model training pipelines concurrently and trillions of...


  • Mountain View, California, United States Google Full time

    About the RoleAs a software engineer at Google, you will have the opportunity to work on cutting-edge technologies that impact billions of users worldwide. Our team is responsible for building the technical foundation behind Google's flagship products, and we are looking for talented engineers to join us.ResponsibilitiesDesign and develop performant,...


  • Mountain View, California, United States Samsung Electronics Perú Full time

    About the RoleSamsung Electronics Perú is seeking a highly skilled Machine Learning Serving Engineer to join our team. As a Machine Learning Serving Engineer, you will play a critical role in designing and developing highly scalable, real-time machine learning infrastructure to support our advertising business.Key ResponsibilitiesDesign and Develop Machine...


  • Mountain View, California, United States Google Full time

    About the RoleWe're seeking a highly skilled Senior Software Engineer to join our Machine Learning team at Google. As a key member of our team, you will be responsible for designing, developing, and deploying cutting-edge machine learning algorithms and models.As a Senior Software Engineer, you will work closely with our team of experts to develop and...


  • Mountain View, California, United States Celebree School of Pike Creek Full time

    About the RoleCelebree School of Pike Creek is seeking a highly experienced Director of Engineering, Machine Learning to lead our team of software engineers and researchers in developing cutting-edge autonomous driving technology. As a key member of our engineering leadership team, you will be responsible for driving the technical vision and strategy for our...


  • Mountain View, California, United States Coupang Full time

    About CoupangCoupang is a leading e-commerce company that is revolutionizing the way people shop, eat, and live. Our mission is to build the future of commerce, and we're looking for talented engineers to join our team.Job SummaryWe're seeking a Senior Staff Engineer to lead the development of our Machine Learning Platform. As a key member of our Data...


  • Mountain View, California, United States Google Full time

    About the RoleAs a Machine Learning Engineer on the Starline audio team at Google, you will play a key role in shaping the future of immersive audio experiences. You will be responsible for training, evaluating, and optimizing AudioLM-based models that push the boundaries of speech enhancement and sound separation.ResponsibilitiesTrain and fine-tune...


  • Mountain View, California, United States IBM Full time

    About the RoleWe are seeking a highly skilled Machine Learning Engineer to join our team at IBM. As a key member of our conversational AI group, you will be responsible for designing, developing, and deploying machine learning models that enable our conversational AI system to understand and respond to user queries in multiple languages.Key...