Current jobs related to Research/Compiler Engineer - Palo Alto, California - Lightning AI

  • Research Engineer

    2 weeks ago


    Palo Alto, California, United States Lightning AI Full time

    About the RoleWe are seeking a highly skilled Research Engineer to join our team at Lightning AI. As a Research Engineer, you will play a key role in the development of our Lightning Thunder compiler, an open-source project developed in collaboration with NVIDIA. Your primary responsibility will be to develop the Thunder compiler using your deep experience...


  • Palo Alto, California, United States Rivian Full time

    About RivianRivian is a company that aims to keep the world adventurous forever. Our team shares a love of the outdoors and a desire to protect it for future generations.Job SummaryWe are seeking a skilled Compiler Engineer to join our ML Compiler team. As a key member of the team, you will work on software tools to enable inference of deep learning networks...


  • Palo Alto, California, United States Tesla Full time

    About the RoleWe are seeking a highly skilled Compiler Engineer to join our AI Inference team at Tesla. As a key member of our team, you will be responsible for designing and developing the compiler for our AI inference stack, which runs neural networks in millions of Tesla vehicles and Optimus.You will collaborate closely with our AI Engineers and Hardware...


  • Palo Alto, California, United States The Pennsylvania State University Full time

    Job DescriptionWe are seeking an undergraduate student researcher to join our Fluids Machinery Department at the Applied Research Laboratory (ARL) at Penn State University.ResponsibilitiesAssist with writing and debugging computer code to analyze turbomachinery and hydrodynamic designsAssist in the development of optimization methods for constrained...


  • Palo Alto, California, United States Rivian Full time

    About RivianRivian is a pioneering company that's revolutionizing the electric vehicle industry. Our mission is to keep the world adventurous forever, and we're looking for talented individuals to join our team.We're a company that's constantly pushing the boundaries of what's possible. We challenge conventional thinking, seek out new solutions, and operate...

  • Research Assistant

    2 months ago


    Palo Alto, California, United States Penn State University Talent Acquisition Full time

    Job DescriptionJoin the Materials Processing Division of the Applied Research Laboratory (ARL) at Penn State University as a highly motivated undergraduate student to assist with various activities in the division.Key ResponsibilitiesAssist with material synthesis, processing, characterization, and device fabricationDevelop and implement process simulations...


  • Palo Alto, California, United States Pennsylvania State University Full time

    Job Summary:The Materials Processing Division of the Applied Research Lab (ARL) at Pennsylvania State University is seeking highly motivated undergraduate students to assist with various activities in the Materials Processing Division.As a Research Assistant, you will work under the guidance of experienced researchers to develop and implement new materials...


  • Palo Alto, California, United States Penn State University Talent Acquisition Full time

    Job Summary:The Materials Processing Division of Applied Research Lab (ARL) is seeking highly motivated undergraduate students to assist with various activities in the division. As a research assistant, you will work on a broad range of programs involving cold spray, sputtering, chemical vapor deposition, physical vapor deposition, hot pressing, plating, and...


  • Palo Alto, California, United States Penn State University Talent Acquisition Full time

    Job Description and Position RequirementsThe Materials Processing Division of the Applied Research Laboratory (ARL) at Penn State University is seeking highly motivated undergraduate students to assist with various activities in the Materials Processing Division.As an authorized DoD SkillBridge partner, ARL welcomes transitioning military members to...

  • AI Research Engineer

    1 month ago


    Palo Alto, California, United States Acceler8 Talent Full time

    Join Acceler8 Talent as a Founding Machine Learning Research EngineerWe are seeking enthusiastic individuals to join our pioneering team as Founding ML Research Engineers. If you're passionate about advancing AI systems and tackling complex challenges in machine learning, we want to hear from you.This role offers both junior and senior opportunities,...


  • Palo Alto, California, United States Penn State University Talent Acquisition Full time

    Job Description and Position RequirementsWe are seeking a highly skilled Systems Research and Development Engineer to join our team at the Applied Research Laboratory (ARL) at Penn State University.The successful candidate will be responsible for providing hands-on leadership of project teams, technical management and leadership of research and development...

  • Software Engineer

    2 weeks ago


    Palo Alto, California, United States Pennsylvania State University Full time

    Job Summary:We are seeking a highly motivated and experienced Software Engineer to join our team at the Applied Research Laboratory (ARL) at Penn State University. The successful candidate will be responsible for designing and developing software solutions for various research projects, including algorithm implementation and integration. The ideal candidate...

  • Software Engineer

    2 weeks ago


    Palo Alto, California, United States Penn State University Talent Acquisition Full time

    Job DescriptionWe are seeking a highly motivated and experienced Software Engineer to join our team at Penn State University's Applied Research Laboratory (ARL). As a key member of our Communications and Signal Processing Division, you will be responsible for designing and developing cutting-edge software solutions to support various research processes and...


  • Palo Alto, California, United States Penn State University Talent Acquisition Full time

    Job Summary:We are seeking a highly motivated and experienced Data Research Engineer to join the Algorithms, Prototyping and Integration (API) Department of the Applied Research Laboratory (ARL) at Penn State University.The successful candidate will assist in providing our customers with state-of-the-art visualization and decision support software-based...


  • Palo Alto, California, United States Stanford Health Care Full time

    Job SummaryWe are seeking a highly skilled and experienced Research Software Engineering Manager to join our team at Stanford Health Care. As a key member of our Research Technology Application Solutions team, you will be responsible for leading and managing a team of software engineers to deliver high-impact software solutions supporting the needs of...


  • Palo Alto, California, United States Tykhe Inc Full time

    Join Our Team as a Lead Research Scientist/EngineerWe are seeking a highly skilled and experienced Lead Research Scientist/Engineer to join our team at Tykhe Inc in Palo Alto, CA. Our company specializes in building cutting-edge GenAI infrastructure, focusing on Voice/Audio/Speech, Vision, and Multi-modal platforms.If you have expertise in designing,...


  • Palo Alto, California, United States Penn State University Talent Acquisition Full time

    Job DescriptionWe are seeking a highly motivated and experienced Systems Research and Development Engineer to join our team in the Communications and Signal Processing (CSP) Division at Penn State University. The successful candidate will work with senior technical staff to research, design, develop, integrate, and test advanced communications systems.Key...

  • AI Research Engineer

    1 month ago


    Palo Alto, California, United States Acceler8 Talent Full time

    About the RoleWe are seeking a highly skilled AI Research Engineer to join our team as a Member of Technical Staff, Research Engineer (Inference). As a key member of our team, you will be responsible for optimizing and deploying large language models (LLMs) for inference across cloud and on-prem environments.Key Responsibilities:Optimize LLMs for inference...


  • Palo Alto, California, United States Acceler8 Talent Full time

    Unlock AI Innovation as a Member of Technical Staff, Research Engineer (Inference)Acceler8 Talent is seeking a highly skilled Research Engineer (Inference) to join our team of AI innovators. As a key member of our technical staff, you will play a pivotal role in optimizing and deploying state-of-the-art models for real-world applications.About the CompanyOur...


  • Palo Alto, California, United States Pennsylvania State University Full time

    Job Opportunity:We are seeking a part-time Research and Development Engineer to join the Antennas and Electromagnetics Department team of the Applied Research Laboratory (ARL) at Penn State.About the Role:This position will conduct EM and antenna analyses, modify and test EM propagation and antenna models, implement various methods from published works for...

Research/Compiler Engineer

2 months ago


Palo Alto, California, United States Lightning AI Full time

About Us

Lightning AI is a pioneering company redefining the development of artificial intelligence products. Our mission is to simplify the lifecycle of machine learning development, making it accessible to individuals and enterprises alike.

Our Values

We are committed to building an inclusive company and developing ethical AI that improves humanity. Our values include:

  • Integrity: We strive to create AI that benefits society as a whole.
  • Perseverance: We empower our community to tackle complex challenges and make a meaningful impact.
  • Curiosity: We are driven to advance the state of AI research and foster its application in innovative products.

Job Summary

We are seeking a research engineer to work on the Lightning Thunder compiler and the PyTorch Lightning stack. This is an opportunity to create groundbreaking technology that will transform the machine learning ecosystem.

Key Responsibilities

  • Develop the Thunder compiler, an open-source project developed in collaboration with NVIDIA, using your expertise in PyTorch, JAX, or other deep learning frameworks.
  • Engage in performance-oriented model optimizations, focusing on distributed training and inference.
  • Develop optimized kernels in CUDA or Triton to target specific use-cases.
  • Integrate Thunder throughout the PyTorch Lightning ecosystem.
  • Engage with the community and champion its growth.
  • Support the adoption of Thunder across the industry.
  • Work closely with the Lightning team as a strategic partner.

Benefits and Perks

We offer competitive base salaries and stock options, as well as a range of benefits, including:

  • Competitive medical, dental, and vision plans.
  • Life and AD&D insurance.
  • Flexible paid time off and winter closure.
  • Generous paid family leave benefits.
  • Meal reimbursement and home office stipend.
  • Learning and development stipend.
  • Citibike membership and gym membership.