Research/Compiler Engineer London, England, United Kingdom

3 weeks ago

Palo Alto CA United States Tbwa ChiatDay Inc Full time

Lightning AI is the company reimagining the way AI is built. After creating and releasing PyTorch Lightning in 2019, Lightning AI was launched to reshape the development of artificial intelligence products for commercial and academic use.

Focusing on simplicity, sustainability, modularity, and extensibility, Lightning AI streamlines the lifecycle of machine learning development to expand widespread AI adoption. Our aim is to enable individual and enterprise users to build deployment-ready AI tools without having to hire experts or sink resources into in-house infrastructure.

Lightning AI has offices in New York City, Palo Alto, and London, and is backed by $58.6 million in funding from Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.

Our Values

We are building a company of people who care deeply about the impact of AI on our society. We value:

  • Integrity: AI is bound to change how we interact with computers, each other, and the world—for the better. We are committed to building an inclusive company and developing ethical AI that improves humanity.
  • Perseverance: Be bold, take the initiative, and make decisions with confidence. We empower our community, clients, and colleagues to tackle the impossible.
  • Curiosity: We are genuinely interested in finding solutions to advance the state of AI research and foster its application in novel products. It's not about whether we can solve the problem, but how.

We also recognize the genuine lack of diversity in our industry, which has to change. One of our goals is to help drive that change. Lightning AI is fully committed to diversity, equity, and inclusion in all of its practices, especially when it comes to growing our team. Our culture promotes inclusion, and we embrace how rewarding it is to work with people from all walks of life.

What we're looking for

We are looking for a software engineer to work directly on the Lightning Thunder compiler and the rest of the PyTorch Lightning stack. This is an opportunity to create groundbreaking technology that will transform the machine learning ecosystem.

What you’ll do
  • Develop the Thunder compiler, an open-source project developed in collaboration with NVIDIA, using your deep experience in PyTorch, JAX, or other deep learning frameworks.
  • Engage in performance-oriented model optimizations, around distributed training as well as 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 within the Lightning team as a strategic partner.
Benefits and Perks

We offer competitive base salaries and stock options with a 25% one year cliff and monthly vesting thereafter. For our international employees, we work with Velocity Global to pay you in your local currency and provide equitable benefits across the globe.

In the US, we offer:

  • 90% monthly premium contributions towards medical and 100% monthly premium contributions towards select dental and vision plans for you and your dependents.
  • Life and AD&D insurance.
  • Flexible paid time off plus 1 week of winter closure.
  • Generous paid family leave benefits.
  • $500 monthly meal reimbursement, including groceries & food delivery services.
  • $1,000 home office stipend.
  • $1,000 annual learning & development stipend.
  • 100% Citibike membership (NYC only).
  • Additional various medical and mental health services.
#J-18808-Ljbffr

  • Fullstack Engineer London, England, United Kingdom

    3 weeks ago

    Palo Alto, CA, United States Tbwa ChiatDay Inc Full time

    Lightning AI is the company reimagining the way AI is built. After creating and releasing PyTorch Lightning in 2019, Lightning AI was launched to reshape the development of artificial intelligence products for commercial and academic use. Focusing on simplicity, sustainability, modularity, and extensibility, Lightning AI streamlines the lifecycle of machine...

  • Research/Compiler Engineer

    1 month ago

    Palo Alto, United States Lightning AI Full time

    Who We Are Lightning AI is the company reimagining the way AI is built. After creating and releasing PyTorch Lightning in 2019, Lightning AI was launched to reshape the development of artificial intelligence products for commercial and academic use. We are on a mission to simplify AI development, making it accessible to everyone-from solo researchers to...

  • Research/Compiler Engineer

    3 weeks ago

    Palo Alto, United States Lightning AI Full time

    Who We Are Lightning AI is the company reimagining the way AI is built. After creating and releasing PyTorch Lightning in 2019, Lightning AI was launched to reshape the development of artificial intelligence products for commercial and academic use. We are on a mission to simplify AI development, making it accessible to everyone-from solo researchers to...

  • Senior Compiler Engineer

    3 weeks ago

    San Francisco, CA, United States Nexus Full time

    We are seeking a highly skilled Senior Compiler Engineer to join our team and play a critical role in pushing the boundaries of performance and efficiency for our decentralized platform. You will leverage your expertise in LLVM and Cranelift to design and implement cutting-edge compiler optimizations for various architectures as we innovate in the world of...

  • ML Compiler Engineer for AI Advancement

    7 days ago

    Palo Alto, California, United States Rivian Full time

    About RivianRivian is committed to preserving the world's adventurous spirit through our mission. Our goal extends to building emissions-free Electric Adventure Vehicles and attracting innovative individuals who share this vision.Our company thrives on challenging conventional wisdom, reframing problems, and exploring uncharted territories. While our...

  • Compiler Engineer for Deep Learning

    2 days ago

    Palo Alto, California, United States Rivian Full time

    About RivianRivian is dedicated to building a more sustainable future through its innovative electric vehicles. The company's mission is centered around preserving the world's adventurous spirit and protecting the environment for future generations.As a leading player in the electric vehicle market, Rivian continuously challenges traditional approaches to...

  • Compiler Specialist

    2 weeks ago

    Palo Alto, California, United States Rivian Full time

    Job DescriptionWe're seeking an experienced MACHINE LEARNING ENGINEER to join our team as a key member of the ML Compiler team. In this role, you will work closely with the ADAS and Hardware teams to develop software tools that enable inference of deep learning networks on Rivian Hardware Platforms.Your primary responsibilities will include:Designing and...

  • Machine Learning Compiler Engineer

    3 weeks ago

    Oregon, IL, United States Qualcomm Full time

    Company: Qualcomm Technologies, Inc. Job Area: Engineering Group, Engineering Group > Machine Learning Engineering General Summary: If you’re interested in advancing and applying mathematics, programming languages theory, and advanced algorithms to program optimization for cutting-edge machine learning accelerators, then you really want to be talking...

  • Workday Financials

    3 weeks ago

    London, KY, United States CloudRock Partners Ltd Full time

    Workday Financials & Adaptive Planning (Managing Consultant) United Kingdom - London Location - London (Hybrid) About the Role This role involves delivering comprehensive solutions to our customers by leveraging extensive knowledge of Workday Financials and Adaptive Planning, ensuring efficient and effective system performance, and driving continuous...

  • Engineering Research

    3 weeks ago

    Palo Alto, CA, United States ScienceCast, Inc Full time

    Job position at Stanford University, Stanford, California . Deadline is on September 16, 2023 Director, Engineering Research Administration Job summary Stanford’s School of Engineering seeks a director of Engineering Research Administration (ERA) to support faculty, staff, and students engaged in cutting edge research aimed at solving the world’s...

  • Software Engineer – Compiler

    3 weeks ago

    Sunnyvale, CA, United States Baidu Full time

    Do you want to be part of the AI revolution? Do you want to think out of the box, thriving on challenges in the AI industry and the desire to solve them? Do you want to work with a world-class team to explore the fast-growing AI hardware opportunities and impact on the AI industry? We’re looking forward to you joining us to collaborate, contribute, and...

  • AI Research and Development Engineer

    7 days ago

    Palo Alto, California, United States OPPO US Research Center Full time

    We are seeking an Advanced Data Science Specialist to join our research and development team at OPPO US Research Center, where we combine technology innovation with business problem-solving expertise. In this role, you will play a crucial part in refining our recommender system, search algorithms, and ad targeting algorithms.Key Responsibilities:Collaborate...

  • Advanced Analytics Senior Analyst

    3 weeks ago

    San Mateo, CA, United States Tbwa ChiatDay Inc Full time

    Advanced Analytics Senior Analyst (Econometrician) United Kingdom, London Why PlayStation? PlayStation isn’t just the Best Place to Play — it’s also the Best Place to Work. Today, we’re recognized as a global leader in entertainment producing The PlayStation family of products and services including PlayStation5, PlayStation4, PlayStationVR,...

  • Research Engineer, Applied ML

    3 weeks ago

    Mountain View, CA, United States Thetalkingmachines Full time

    Send me a weekly email with the latest jobs. Research Engineer, Applied ML (Mountain View) Job Type: Full Time Location: Mountain View, CA The DeepMind Applied group collaborates with teams at Google and elsewhere to apply our cutting-edge research to products and infrastructure used by many millions of people across the world. We’re very proud to already...

  • Compiler Engineer II

    3 weeks ago

    Cupertino, CA, United States Amazon Full time

    The AWS Neuron Compiler team is actively seeking skilled compiler engineers to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX. Your role...

  • Plant Maintenance Engineer

    3 weeks ago

    Boston, MA, United States Randstad Cpe London Full time

    PLANT MAINTENANCE MECHANICAL ENGINEER - LONDON - 3/MONTH CONTRACT - £38 PER HOUR (UMBRELLA) INSIDE IR35 ARM is working with a leading transport company, involved with some large-scale construction in London. We are currently recruiting for a Plant Maintenance Mechanical Engineer, based in London.About You: You will have had experience and knowledge working...

  • Research Engineer

    3 months ago

    palo alto, United States Acceler8 Talent Full time

    Join US as a Founding Machine Learning Research EngineerAre you passionate about advancing AI systems and tackling complex challenges in machine learning? We are seeking enthusiastic individuals to join our pioneering team as Founding ML Research Engineers. This role offers both junior and senior opportunities, allowing individuals at different stages of...

  • Research Engineer

    7 months ago

    Palo Alto, United States Acceler8 Talent Full time

    Join US as a Founding Machine Learning Research EngineerAre you passionate about advancing AI systems and tackling complex challenges in machine learning? We are seeking enthusiastic individuals to join our pioneering team as Founding ML Research Engineers. This role offers both junior and senior opportunities, allowing individuals at different stages of...

  • Research Engineer

    2 months ago

    palo alto, United States Acceler8 Talent Full time

    Join US as a Founding Machine Learning Research EngineerAre you passionate about advancing AI systems and tackling complex challenges in machine learning? We are seeking enthusiastic individuals to join our pioneering team as Founding ML Research Engineers. This role offers both junior and senior opportunities, allowing individuals at different stages of...

  • Machine Learning Engineer London, United Kingdom

    3 weeks ago

    Holiday, FL, United States Datatonic Ltd. Full time

    At Datatonic, we are Google Cloud's premier partner in AI, driving transformation for world-class businesses. We push the boundaries of technology with expertise in machine learning, data engineering, and analytics on Google Cloud. By partnering with us, clients future-proof their operations, unlock actionable insights, and stay ahead of the curve in a...