Skip to content
xAIxAIPalo Alto, CA

Member of Technical Staff

Hands-on technical leader building and scaling large language models and AI systems. Requires 3-5+ years of AI/ML experience with strong Python and deep learning frameworks.

324k – 396k/yr
On-site5+ YOEML Engineering

About the role

Responsibilities

  • Build collaborative relationships with the Principal and Staff engineering community as well as with engineering and product management leaders, and partner to deliver impact.
  • Define the vision and strategy for the organization and have a substantial impact on the vision and strategy of customer and partner organizations.
  • Plan and deliver projects that impact multiple organizations, including models and large language model training, pipeline parallelism training of large language models, and fine tuning large language models with truthful data.
  • Identify opportunities for technological differentiation, investment, or divestment.
  • Ensure organization’s work is aligned with broader company objectives.
  • Introduce innovative techniques and analyses to the AI field to facilitate breakthroughs in quantitative reasoning and language understanding.
  • Provide mentorship and guidance to senior technical leaders and managers.
  • Work on hands-on technical problems including design and implementation.
  • Perform cutting-edge research on advanced techniques from AI and deep learning, including neural network architectures, language modeling, and speech recognition.
  • Work closely with leaders across the company to deliver impactful projects which may involve work in areas such as machine learning, applied data science, recommendation systems, and information retrieval systems.

Requirements

  • Bachelor’s degree in Computer Science, Artificial Intelligence, Machine Learning, Information Technology, or a related field, plus 5 years of progressive post-baccalaureate experience in AI/ML research or development; OR Master’s degree in Computer Science, Artificial Intelligence, Machine Learning, Information Technology or a related field, plus 3 years of experience in AI/ML research or development.
  • 3 years of experience in each of the following:
    • Developing and applying AI models, including large language models, neural networks, or reinforcement learning.
    • Experience in Python and other relevant languages (such as C++, Java), using ML frameworks (such as PyTorch, TensorFlow, or JAX).
    • Assessing system performance and designing scalable, efficient solutions for inference and model deployment.
    • Optimizing computational efficiency and memory usage through advanced algorithmic techniques or quantization.
    • Building and maintaining robust, high-availability infrastructure for AI/ML services.
    • Collaborating with cross-functional teams to define technical vision, strategy, and deliver impactful projects.
    • Mentoring and guiding senior technical leaders and managers in hands-on technical problem solving.
    • Driving innovation by introducing new techniques and analyses to advance AI capabilities in quantitative reasoning and language understanding.
    • Leading efforts in resource management, automation, and orchestration for large-scale ML infrastructure.

Compensation

  • $324,000 - $396,000 per year

Skills

PythonC++JavaPyTorchTensorFlowJAXLLMsNeural NetworksReinforcement LearningModel DeploymentQuantizationML Infrastructure

Similar roles

ML Engineering jobs
xAI

Member of Technical Staff

xAIPalo Alto, CA

Member of Technical Staff at xAI performing cutting-edge AI research and hands-on engineering to advance large language models, quantitative reasoning, and language understanding. Requires Bachelor's in CS/ML-related field plus 2 years experience with big data tools, Kubernetes, deep learning, NLP/CV/speech, and languages like Rust/C++/Python.

324k – 396k/yr
On-site2+ YOEML Engineering
Decagon

Staff Research Engineer

DecagonSan Francisco, CA +1

Staff Research Engineer leads development of conversational AI models for enterprise support agents, owning multi-quarter initiatives from research to production deployment. Requires 8+ years in AI/ML with experience deploying LLMs, Python fluency, and cross-functional leadership.

325k – 425k/yr
On-site8+ YOEML Engineering
Anthropic

Staff + Senior Software Engineer, Cloud Inference Launch Engineering

AnthropicSan Francisco, CA

Build and own validation pipelines, CI/CD infrastructure, and platform integrations to launch frontier models and inference features reliably across AWS, GCP, and Azure. Requires strong large-scale distributed systems experience and track record improving release velocity.

320k – 485k/yr
Hybrid7+ YOEML Engineering
Anthropic

Staff Software Engineer, Inference

AnthropicSan Francisco, CA +2

Build and maintain distributed inference systems serving Claude to millions of users. Design intelligent routing, autoscaling, and high-performance infrastructure across diverse AI accelerators.

320k – 485k/yr
Hybrid7+ YOEML Engineering
Attentive

Staff Software Engineer, Machine Learning

AttentiveNew York, NY +1

Builds, scales, and operates production-grade ML systems for real-time personalization on Attentive's platform. Requires 6+ years experience with Python, PyTorch/TensorFlow, and scalable ML pipelines in a fast-paced environment.

320k – 360k/yr
Remote6+ YOEML Engineering