Member of Technical Staff - RL Training Framework

3 hours, 17 minutes ago
Full-time
Lead
Software Development
x.ai

x.ai

x.ai is a technology company founded in 2014 that specializes in developing artificial intelligence personal assistants. Their flagship product is a personal assistant that efficiently schedules meetings for users. With a focus on sustainable business ...

Internet Software & Services
51-250
Founded 2023
$44M raised

Description

  • Design and implement the systems that back all RL workloads at SpaceXAI, from small-scale ablations to production training runs.
  • Profile, debug, and optimize end-to-end training performance.
  • Improve the scalability and observability of the RL stack.

Requirements

  • Experience building, debugging, and optimizing large-scale distributed systems.
  • Comfortable diving into unfamiliar areas and solving problems at all levels of the stack.
  • Proficiency in Python, Jax, Rust, and/or C++.
  • Experience with large-scale LLM training infrastructure (preferred).
  • Strong knowledge of reinforcement learning techniques (preferred).
  • Experience with RL numerics (preferred).

Benefits

  • $180,000 - $440,000 USD base salary.
  • Equity as part of the total rewards package.
  • Comprehensive medical, vision, and dental coverage.
  • Access to a 401(k) retirement plan.
  • Short- and long-term disability insurance.
  • Life insurance.
  • Various other discounts and perks.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior Machine Learning Engineer

ALT 51-250 Capital Markets

Alt is hiring a Senior Machine Learning Engineer to own the end-to-end productization of pricing and underwriting models for its trading-card marketplace, powering real-time valuation and cash advance decisions at scale.

AWS CI/CD Feature Engineering LLM MLflow Python Scikit-learn SQL
3 hours, 2 minutes ago

Senior Researcher, AI Civics

Data & Society 11-50 Professional Services

Data & Society is hiring a Senior Researcher, AI Civics to lead participatory research and partnership-driven co-design work for a two-year program focused on civic action and community control over AI.

3 hours, 17 minutes ago

Senior Engineer, Machine Learning Platform

Super Technologies Pvt Ltd 1-10 aviation & aerospace

Super Technologies is hiring a Senior Engineer for its ML Platform squad to build a new machine learning platform that enables ML and AI adoption across the engineering organization and supports large-scale data-driven experiences.

AWS Azure Docker Flink GCP Kafka Kubernetes Machine Learning MLflow MLOps Python
3 hours, 32 minutes ago

Staff Machine Learning Engineer

Boulevard 251-1K Diversified Consumer Services

Boulevard is hiring a Staff ML Engineer to help build and scale AI and machine learning solutions on its modern data platform for conversational analytics, automation, and customer-facing experiences.

CI/CD Databricks Feature Engineering LLM Machine Learning MLOps Python Snowflake SQL
3 hours, 32 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers