Member of Technical Staff

at xAI
📍 Palo Alto, United States
USD 324,000-396,000 per year
MIDDLE
✅ On-site

Used Tools & Technologies

LLM

Required Skills & Competences

Python @ 3 Java @ 3 Machine Learning @ 3 Data Science @ 3 TensorFlow @ 3 Communication @ 6 Mentoring @ 3 Prioritization @ 6 Product Management @ 3 PyTorch @ 3 Deep Learning @ 3 AI @ 3 Reinforcement Learning @ 3 JAX @ 3

Details

xAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. The team is small, highly motivated, and focused on engineering excellence. The organization expects hands-on contribution, strong prioritization, and clear communication across the company.

Responsibilities

  • Build collaborative relationships with Principal and Staff engineering, engineering leaders, and product management to deliver impact.
  • Define vision and strategy for the organization and influence customer and partner organization strategy.
  • Plan and deliver projects that impact multiple organizations, including work on models and large language model training, pipeline-parallel training of large language models, and fine-tuning large language models with truthful data.
  • Identify opportunities for technological differentiation, investment, or divestment and ensure alignment with broader company objectives.
  • Introduce innovative techniques and analyses to advance AI capabilities in quantitative reasoning and language understanding.
  • Provide mentorship and guidance to senior technical leaders and managers, and participate in hands-on design and implementation.
  • Perform research and engineering on advanced AI and deep learning techniques, including neural network architectures, language modeling, and speech recognition.
  • Work closely across the company on projects in machine learning, applied data science, recommendation systems, and information retrieval systems.
  • Lead efforts in resource management, automation, and orchestration for large-scale ML infrastructure and ensure high-availability AI/ML services.

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 a Master’s degree in a related field plus 3 years of AI/ML research or development experience.
  • Minimum 3 years of experience in each of the following areas:
    • 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) and 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 using 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 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 in quantitative reasoning and language understanding.
    • Leading resource management, automation, and orchestration for large-scale ML infrastructure.

Compensation

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

How to Apply

Any interested applicant may click the APPLY NOW button on the original posting to apply. Reference: 00101156.