SpaceXAI’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, operating with a flat organizational structure. All employees are expected to be hands‑on, contribute directly to the company’s mission, and exhibit strong communication, prioritization, and initiative.
Member of Technical Staff at SpaceXAI
On-site - Palo Alto, CA, USA
More jobs at SpaceXAISalary
USD 324,000 - 396,000/year
Requirements
Skills
- Bachelor’s degree in Computer Science, Artificial Intelligence, MachineLearning, Information Technology, or a related field
- 5 years of progressive post-baccalaureate experience in AI/ML research or development
- Alternatively, Master’s degree in Computer Science, Artificial Intelligence, Machine Learning, Information Technology or a related field
- 3 years of experience in AI/ML research or development
- 3 years of experience developing and applying AI models, including large language models, neural networks, or reinforcement learning
- Experience in Python and other relevant languages such as C++ and Java
- Experience using ML frameworks such as PyTorch, TensorFlow, or JAX
- Experience assessing system performance and designing scalable, efficient solutions for inference and model deployment
- Experience optimizing computational efficiency and memory usage through advanced algorithmic techniques or quantization
- Experience building and maintaining robust, high-availability infrastructure for AI/ML services
- Experience collaborating with cross-functional teams to define technical vision, strategy, and deliver impactful projects
- Experience mentoring and guiding senior technical leaders and managers in hands‑on technical problem solving
- Experience driving innovation by introducing new techniques and analyses to advance AI capabilities in quantitative reasoning and language understanding
- Experience leading efforts in resource management, automation, and orchestration for large-scale ML infrastructure
Responsibilities
- Build collaborative relationships with the Principal and Staff engineering community and engineering and product management leaders
- Partner to deliver impact and define the vision and strategy for the organization
- Plan and deliver projects that impact multiple organizations, including models, large language model training, pipeline parallelism, and fine‑tuning 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 for 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 in machine learning, applied data science, recommendation systems, and information retrieval systems
Technologies
PythonC++JavaPyTorchTensorFlowJAX
See if your resume is ready for this job
See how our AI can optimize your resume and improve your chances for this role.