Persimmons, Inc. · 4 months ago
Compiler Architect
San Jose, CA
Full-time
Onsite
Senior Level
6+ years expPersimmons, Inc. is building the infrastructure for the next decade of AI, founded by veteran technologists. They are seeking a Compiler Architect to lead the optimization of their Persimmons Compiler, focusing on converting AI models into assembly code and developing novel scheduling algorithms.
Computer Hardware
Responsibilities
Design and build the compiler that converts AI models from popular ML frameworks into assembly code that runs on our accelerator hardware
Develop and implement novel scheduling algorithms that push the boundaries of technology
Collaborate with cross-functional teams to design, test, and optimize our hardware and software solutions
Analyze and improve the efficiency, scalability, and performance of our systems
Stay abreast of industry trends and advancements to ensure our solutions remain competitive and innovative
Provide technical leadership across the compiler team, mentoring engineers in advanced compiler techniques, and help scale the team as the company grows
Qualification
Required
6+ years of experience in compiler development, and deep knowledge of modern compiler frameworks (LLVM, MLIR, TVM, XLA, IREE)
Proven track record of leading compiler or runtime systems projects from design through deployment
Experience processing models from popular frameworks (e.g. PyTorch, TensorFlow, JAX), and familiarity with model architecture and workloads (transformers, diffusion models, etc.)
Familiarity with auto-scheduling and program synthesis techniques for high-performance ML kernels (e.g., TVM, Halide, or Ansor), or experience with other hardware-aware scheduling techniques
Familiarity with hardware architectures and their optimization implications, including memory hierarchies, systolic arrays, DMA engines, and GPU-style parallelism
If you can do the above, you already have strong C++ and python skills
BS/MS/PhD degree in Computer Science, Computer Engineering, or related field (or equivalent experience)
Strong interpersonal, verbal and written communications skills
Capability to achieve objectives under tight deadlines
Experience executing tasks while managing competing priorities
Practical knowledge working with large code bases
Experience writing and debugging multithreaded programs
Deep understanding of technology and passion for what you do
Strong teamwork, specifically a proven ability to effectively guide and influence within a dynamic matrix environment
Excellent problem-solving skills and the ability to work in a dynamic, fast-paced environment
Benefits
Competitive salary and benefits package
Flexible PTO
401k
Company
Persimmons, Inc.
From custom silicon to intelligent algorithms, we’re breaking through the bottlenecks holding AI back, delivering orders-of-magnitude more performance, anywhere it’s needed.
11-50 employees
Funding
Current Stage
Early StageRecent News
2024-11-07
Company data provided by crunchbase