Jobs
>
Los Angeles

    Principal Engineer, Frameworks - Los Angeles, United States - NVIDIA

    Default job background
    Description


    We are hiring a Principal Engineer to oversee developing our AI platforms, with a heavy emphasis on large language models (LLMs) such as PAXML.

    Work with a dedicated group of engineers in a fast-paced production environment to make a tangible impact to Deep Learning Frameworks, particularly JAX.

    Come join us in these exciting times and make a sizable difference in the exploding world of AI As an all-stack engineer with a focus on the back-end libraries and compilers, you will play a meaningful role in NVIDIA's efforts in contributing to the Deep Learning ecosystem.

    What You Will Be Doing

    Oversee and lead design and development of Deep Learning platform solutions with our Google partners
    Design and build core Frameworks and Compilers and drive peak performance on NVIDIA GPUs
    Evaluate and improve the end-to-end performance of generative AI models, especially large-language models (e.g. PAXML), on NVIDIA hardware
    Work with AI applied researchers and leaders to build highly-efficient ML models
    Build tools that will increase the efficiency of teams developing AI-based systems.

    Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products.

    Impact the design of next generation of accelerated computing software and hardware
    What We Need To See

    BS in Computer Science or Computer Engineering or related field (or equivalent experience)
    15+ years relevant experience
    Background in C++, Python, MLIR, LLVM
    Experience with at least one machine learning framework, e.g. JAX, PyTorch, TensorFlow
    Familiar with such concepts as asynchronous computing, sharding, collectives, various flavors of parallelism, GPU kernels, ML building blocks (e.g. Multi-Head Attention, MLP)
    Proven technical foundation in GPU architectures, numeric libraries, modular software design.
    Highly motivated with excellent verbal and written communication skills.
    Ability to work collaboratively with and lead multi-functional teams. Coordinates effectively across organizational boundaries.
    Ways To Stand Out From The Crowd

    Prior experience in MLIR, XLA, JAX, LLVM, and CUDA
    Firm grasp of distribution computation such as data, model, ZeRO, and pipeline parallelism
    Being intimately familiar with Google C++ Style Guide and having contributed to OSS
    Collaborative and production-minded demeanor
    The base salary range is $268,000 - $414,000. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

    You will also be eligible for equity and benefits .NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer.

    As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

    #J-18808-Ljbffr


  • Apple Los Angeles, United States Full time

    Summary · Apple is where individual imaginations gather together, committing to the values that lead to great work. Every new product we build, service we create, or Apple Store experience we deliver is the result of us making each other's ideas stronger. That happens because eve ...


  • HIKINEX Pasadena, CA, United States

    Job Responsibilities: · Enterprise Applications Software Engineer who shall be responsible for the development of Product Lifecycle Management applications using the current Oracle ADF, Oracle Business Intelligence Reports, and SOA technologies. The Enterprise Applications Softw ...


  • Fitt Talent Partners Los Angeles, United States

    **Fitt Talent Partners is a specialized recruitment firm working with top health, fitness, and wellness companies.** · **We're filling this role for a client.** · **Responsibilities**: · - Analytics Instrumentation Strategy: · - Develop and execute a comprehensive analytics instr ...

  • BambuMeta

    Junior Developer

    3 days ago


    BambuMeta Los Angeles, United States

    **Responsibilities**: · - Assist with database management and development, including SQL and MySQL. · - Collaborate with the development team to design and implement new features and enhancements. · - Provide support for DevOps processes, including AWS administration, code pushes ...


  • Divergent 3D Los Angeles, United States

    Divergent is a technology company that has architected, invented, built, and commercialized an end-to-end factory system called the Divergent Adaptive Production System (DAPS) that comprehensively uses machine learning to optimally engineer, additively manufacture, and flexibly a ...


  • System1 Los Angeles, United States

    System1 is looking for a **Director of Engineering** to be the Lead of our Platform Monetization team. You will design, build, operate, debug, optimize, and deploy highly-distributed and low-latency web-based systems with a growing team of talented engineers. · In this role, you ...


  • California FAIR Plan Association Los Angeles, United States

    *** · The Application Developer Supervisor is the primary point of contact on implementation projects for all internal Duck Creek project teams and is recognized by the project team as the expert on the product and project. · **PRINCIPAL DUTIES AND RESPONSIBILITIES** · - Conduct ...


  • Primesoftinc Los Angeles, United States

    **Role**:Front-End Web Developer · **Exp**: 10+ Years · **Location**: 5999 Center Dr, Los Angeles, CA Weekly 3 days onsite) · Independents and H1 only (no EAD s) · **MOI**: Phone + Skype · **Skills Required**: · - **Strong front-end web development experience (role is 80% front-e ...

  • LEGALMATION

    React Js Developer

    2 days ago


    LEGALMATION Los Angeles, United States

    **Job Title: React JS TypeScript Developer** · **Company Overview**: · **Position Overview**: · **Responsibilities**: · - Develop responsive user interfaces using · - Implement reusable components and front end libraries for future use. · - Ensure the technical feasibility of UI ...


  • UCLA Los Angeles, United States

    The Supervisor, Endpoint Solutions will be responsible for managing and supervising a team of high-performing endpoint support analysts who provide technical support for endpoint devices, such as laptops, desktops, tablets, and mobile phones. The Supervisor, Endpoint Solutions wi ...


  • PERFORMING ARTS CENTER Los Angeles, United States

    **Program Manager, Civic Strategies & Partnerships **TMC Arts**: · As L.A.'s premiere performing arts destination, The Music Center convenes artists, communities, and ideas with the goal of deepening the cultural lives of every resident in Los Angeles County. The $70 million non- ...


  • MedInformatix (Medinformatix) Los Angeles, United States

    **Essential Duties and Responsibilities**: · Maintain existing code and functionality once released in production · Maintain hands-on knowledge of and current best practices for standards-based technologies · Assist in the design, development & management of new modules and tools ...


  • TikTok Los Angeles, United States

    Responsibilities · TikTok is the leading destination for short-form mobile video. At TikTok, our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, and its offices include New York, London, Dublin, Paris, Berlin, Dubai, ...


  • PRICE WATERHOUSE COOPERS Los Angeles, United States

    A career within Functional and Industry Technologies services will provide you with the opportunity to build secure and new digital experiences for customers, employees, and suppliers. We focus on improving apps or developing new apps for traditional and mobile devices as well as ...


  • Fox Corporation Los Angeles, United States

    OVERVIEW OF THE COMPANY · Fox Corporation · Under the FOX banner, we produce and distribute content through some of the world's leading and most valued brands, including: FOX News Media, FOX Sports, FOX Entertainment, FOX Television Stations and Tubi Media Group. We empower a div ...

  • DLR Group

    Pursuit Manager

    5 hours ago


    DLR Group Los Angeles, United States

    We are an integrated design firm with a promise to elevate the human experience through design. This fuels the work we do around the world and inspires our mission to improve the lives of our clients, our communities, and our planet. If this sparks your interest, you're in the ri ...

  • NextPhase

    Salesforce BSA

    6 days ago


    NextPhase Los Angeles, United States

    WHO WE ARE: · is a Data Management and Business Systems solution provider focused on services for enterprises in the cloud. We are leading the innovation agenda with our clients by delivering a portfolio of services to help them improve data quality and optimize systems for bette ...


  • Tri-Force Consulting Services, Inc. Los Angeles, United States

    Title : Senior Data Engineer/Developer · Location : Superior Court of California, County of Los Angeles` · Client : Superior Court of California · Note : This is a remote position; however, they may have to travel on site or to other courts or conferences on rare occasions. · Jo ...


  • Spectrum Industries Lynwood, United States

    Job Description: Compensation: starting at EG 4 + allowance based on experience and qualifications Position within the framework of temporary employment We are currently looking for employees (m/f/d) with completed vocational training or relevant professional experience in the fi ...

  • Bridgewater Consulting Group

    Test Architect

    5 hours ago


    Bridgewater Consulting Group Rosemead, United States

    **Position: Test Architect** · **Location: Hybrid - Rosemead, CA** · **Position Overview**: · We are seeking a highly skilled and experienced Test Architect to join our team. The Test Architect will be responsible for designing testing solutions that effectively assess software s ...