Principal Machine Learning Engineer - Palo Alto - A1

    A1
    A1 Palo Alto

    4 days ago

    Description

    Overview

    A1 is building a proactive AI system that understands context across conversations, plans actions, and carries work forward over time.

    You will be responsible for turning research direction into working, production-grade ML systems. This role owns the execution layer of A1's intelligence – training pipelines, inference systems, evaluation tooling, and deployment.

    Focus

    • Build and own end-to-end ML pipelines spanning data, training, evaluation, inference, and deployment.
    • Fine-tune and adapt models using state-of-the-art methods such as LoRA, QLoRA, SFT, DPO, and distillation.
    • Architect and operate scalable inference systems, balancing latency, cost, and reliability.
    • Design and maintain data systems for high-quality synthetic and real-world training data.
    • Implement evaluation pipelines covering performance, robustness, safety, and bias, in partnership with research leadership.
    • Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies.
    • Collaborate closely with application engineering to integrate ML systems cleanly into backend, mobile, and desktop products.
    • Make pragmatic trade-offs and ship improvements quickly, learning from real usage.
    • Work under real production constraints: latency, cost, reliability, and safety

    Requirements

    • Strong background in deep learning and transformer-based architectures.
    • Hands-on experience training, fine-tuning, or deploying large-scale ML models in production.
    • Proficiency with at least one modern ML framework (e.g. PyTorch, JAX), and ability to learn others quickly.
    • Experience with distributed training and inference frameworks (e.g. DeepSpeed, FSDP, Megatron, ZeRO, Ray).
    • Strong software engineering fundamentals – you write robust, maintainable, production-grade systems.
    • Experience with GPU optimization, including memory efficiency, quantization, and mixed precision.
    • A bias toward shipping, learning fast, and improving systems through iteration.

    Ideal Experience

    • Experience with LLM inference frameworks such as vLLM, TensorRT-LLM, or FasterTransformer.
    • Contributions to open-source ML or systems libraries.
    • Background in scientific computing, compilers, or GPU kernels.
    • Experience with RLHF pipelines (PPO, DPO, ORPO).
    • Experience training or deploying multimodal or diffusion models.
    • Experience with large-scale data processing (Apache Arrow, Spark, Ray).

    Compensation Range

    $347K – $490K

    How We Work

    Our organization is very flat and our team is small, highly motivated, and focused on engineering and product excellence. All members are expected to be hands-on and to contribute directly to the company's mission.

    If there appears to be a fit, we\'ll reach to schedule 3, but no more than 4 interviews.

    Applications are evaluated by our technical team members. Interviews will be conducted via virtual meetings and/or onsite.

    We value transparency and efficiency, so expect a prompt decision. If you\'ve demonstrated the exceptional skills and mindset we\'re looking for, we\'ll extend an offer to join us. This isn\'t just a job offer; it\'s an invitation to be part of a team that\'s bringing AI to have practical benefits to billions globally.


    #J-18808-Ljbffr

  • Only for registered members Palo Alto, CA

    We create breakthrough solutions that make complex tasks feel effortless, teamwork more natural, and ideas more impactful—always with a human-centric mindset. · ...

  • Only for registered members Palo Alto, CA

    We are looking for a world-class Deep Learning Software Engineer who is excited to operate at the forefront of computer vision and deep learning applied to challenging, real-world use-cases. You will be working in a fast-paced environment, surrounded by experts who are passionate ...

  • Only for registered members Palo Alto, CA

    We partner directly with customers to build AI systems that enhance the safety and reliability of their complex applications. · ...

  • Only for registered members Palo Alto, CA

    +We are seeking a Senior Machine Learning Architect who is deeply hands-on, technically rigorous, and excited to own the design, architecture, and delivery of real-world ML systems. · +Architect and build production-grade ML systems end to end, from experimentation through deploy ...

  • Only for registered members Palo Alto, CA

    We're hiring our first Machine Learning Engineer in the United States, a foundational role that will shape how Abaka builds, trains, and optimizes multimodal AI systems. · ...

  • Only for registered members Palo Alto, CA

    We're hiring our first Machine Learning Engineer in the United States, a foundational role that will shape how Abaka builds, trains, and optimizes multimodal AI systems. · Design, build, and optimize scalable machine learning pipelines for multimodal model training, · fine-tuning ...

  • Only for registered members Palo Alto, California, United States

    +The Perception team is looking for a machine learning engineer to develop cutting-edge Computer Vision modules to enhance the on-board perception of our robotic fleet. · +Design and develop in-cabin Computer Vision ML modules for safety critical perception and monitoring applica ...

  • Only for registered members Palo Alto, CA

    We're ALTEN Technology USA an engineering company helping clients bring groundbreaking ideas to life. As part of the global ALTEN Group we deliver across the entire product development cycle from consulting to full project outsourcing. · We take pride in fostering a culture where ...

  • Only for registered members Palo Alto

    +Job summary · We help the world run better · At SAP, we keep it simple: you bring your best to us, and we'll bring out the best in you. · +ResponsibilitiesDevelop and maintain backend services, APIs, and data pipelines in Go, Java, or Python. · Support the optimization of databa ...

  • Only for registered members Palo Alto

    We help teams stay aligned, make better decisions and move faster together. · ...

  • Only for registered members Palo Alto, CA

    Luma AI is a full-stack AI lab building Multimodal AGI. To truly understand the world, models must learn from audio, video, and images. · ...

  • Only for registered members Page Mill Rd Palo Alto, California, United States

    At Diffbot we believe that access to structured information will be the critical resource for the coming wave of intelligent applications. · ...

  • Only for registered members Palo Alto, CA

    The mission of the Digital Intelligence team is to leverage large-scale computation, substantial datasets, and machine learning to enhance our most crucial customer products across a broad spectrum. · We manage a wide array of impactful products and services, reaching countless c ...

  • Only for registered members Palo Alto, California, United States

    We're looking for a machine learning engineer to develop cutting-edge Computer Vision modules to enhance the on-board perception of our robotic fleet. · ...

  • Only for registered members Palo Alto Full time $100,000 - $125,000 (USD)

    We're ALTEN Technology USA an engineering company helping clients bring groundbreaking ideas to life—from advancing space exploration and life-saving medical devices to building autonomous electric vehicles. · ...

  • Only for registered members Palo Alto

    We are looking for a world-class Deep Learning Software Engineer who is excited to operate at the forefront of computer vision and deep learning applied to challenging, real-world use-cases.You will be working in a fast-paced environment, surrounded by experts who are passionate ...

  • Only for registered members Palo Alto

    This is a zero-to-one role with real technical ownership working closely with experienced founders and a small senior engineering team. · Design build and deploy production-grade ML systems solving real-world problems · Select the right technical approach per problem including fi ...

  • Only for registered members Palo Alto $106,900 - $229,400 (USD)

    We help the world run betterAt SAP, we keep it simple: you bring your best to us, and we'll bring out the best in you. · The work is challenging but it matters. You'll find a place where you can be yourself, prioritize your wellbeing, and truly belong. · We are seeking a motivat ...

  • Only for registered members Middlefield Rd, Palo Alto, CA , USA

    We are looking for talented and creative people to join us in using AI-driven technology to make MRIs faster, better, and more affordable. · ...

  • Only for registered members Palo Alto

    This role is designed for an early career machine learning engineer who wants to work directly on real products, · This engineer will help bring intelligent features to life by exploring modern AI techniques, experimenting quickly, and turning promising ideas into production read ...

  • Only for registered members Palo Alto $189,721 - $332,012 (USD)

    We're on a mission to bring everyone the inspiration to create a life they love, and that starts with the people behind the product. Discover a career where you ignite innovation for millions, transform passion into growth opportunities, · celebrate each other's unique experience ...

Jobs
>
Principal machine learning engineer
>
Jobs for Principal machine learning engineer in Palo Alto