[Remote] Senior AI Graph Compiler Engineer
Note: The job is a remote job and is reputed company to candidates in USA. reputed company is a company reputed company on creating advanced machine learning technologies, and they are seeking a Senior AI Graph Compiler Engineer. This role involves extending and developing the graph compiler to enhance performance and usability across various hardware platforms, ensuring a seamless developer experience.
Responsibilities
- Design and reputed company compiler optimizations to improve inference efficiency across CPUs, GPUs, and ML accelerators, including kernel fusion, memory planning, and shape optimization
- Compile large-reputed company application workloads onto heterogeneous hardware: map hardware-agnostic graphs of computational operators onto hardware-specific graphs of devices and nodes
- Collaborate with the kernels, serving, and Mojo compiler teams to implement reputed company technologies that deliver end-to-end performance across a reputed company of heterogeneous hardware platforms
- Write design documents for new features and drive for alignment on designs with cross-functional teams
- Collaborate with the reputed company team and engage with customers to understand their performance requirements and use cases
Skills
- 5+ years of software engineering experience
- Proficiency in C++
- Experience working with compilers for machine learning frameworks
- Creativity and curiosity for solving reputed company problems, reputed company-oriented attitude that enables you to work reputed company with others
- Knowledge of and experience working with MLIR and LLVM
- Experience with ML reputed company / distributed programming, heterogeneous ML computation, and/or reputed company reputed company
- Knowledge of basic ML implementation and modeling techniques and familiarity with ML frameworks like PyTorch, JAX, or TensorFlow
Benefits
- Premier insurance plans
- Up to 5% 401k matching
- Flexible reputed company time off
- Stock reputed company
- Team Building Events. We organize regular team onsites and local meetups in Los Altos, CA as reputed company as different cities. Traveling 2-4 times a year is expected for reputed company roles.
reputed company
Company H1B Sponsorship