Character AI

Software Engineer, Kernels & Performance

Palo Alto
348 days ago

Share:

About us

Character’s mission is to empower everyone with AGI. Our vision is to enable people with our technology so that they can use Character.AI any moment of any day.

Achieving our mission will require solving ambitious technical challenges, including engineering, research, and design, and we are assembling a world-class team to do so. Our founding team includes AI pioneers from Google Brain and Meta Research whose research has led to major breakthroughs in natural language understanding and dialog applications such as Transformers and Google LaMDA.

Check out our beta to get a glimpse into the future.

About the role

We’re looking for a seasoned Research Engineer with expertise in GPU computing to design high-performance kernels for our training and inference workloads using CUDA/CUTLASS, C++, and Python.

Responsibilities:
  • Using the capabilities of GPUs and other accelerators to the fullest extent to make our custom model architectures fast and efficient for training and inference

  • Deliver and maintain high performance GPU and communication kernels that increase utilization and also work well in a distributed training and inference environment

  • Collaborating closely with ML Engineers and Researchers to develop new architectures and algorithms that are aware of I/O & Hardware constraints

  • Communicating with hardware vendors to advise on design of software or future hardware, while also making sure we are keeping up to date with the cutting edge

  • Working on quantization or any form of low-precision arithmetic to increase throughput without degradation in performance

Requirements:
  • Strong C/C++ and Python coding skills

  • Deep understanding of GPUs or other accelerators, along with the ability to effectively profile and analyze existing or new kernels

  • Experience with CUDA/CUTLASS (and experience with Triton-like compilers is a plus)

  • 3+ years of relevant industry experience and experience working with hardware developers

  • MS/PhD in Computer Science and Engineering with a specialization in Computer Architecture, Parallel Computing, Compilers or other System

Character is an equal opportunity employer and does not discriminate on the basis of race, religion, national origin, gender, sexual orientation, age, veteran status, disability or any other legally protected status. We value diversity and encourage applicants from a range of backgrounds to apply.

Please mention that you found this job on MoAIJobs, this helps us grow, thanks!

Related Jobs

Welocalize
Senior Software Engineer
Barcelona, Spain
Helsing
Software Engineer - Backend
Warsaw - Tallinn
Anthropic
Software Engineer, Horizons
San Francisco, CA | New York City, NY | Seattle, WA
Codeium
Software Engineer
Mountain View (HQ)
Palantir
Embedded Software Engineer
Washington, D.C.