AI Model Compiler Engineer

소개

페블스퀘어는 PIM아키텍처를 기반으로 AI 반도체 설계와 AI 솔루션을 개발하는 팹리스 기업입니다. 폰 노이만(Von Neumann)구조의 한계를 극복하기 위해 고성능/초저전력의 PIM기반 AI 반도체 를 성공적으로 양산하고, 멀티 코어 AI 반도체를 개발하여 국내외에서 실증 중이며 AI 반도체 상용화와 활용 촉진을 위해 다양한 AI 솔루션을 개발 중입니다.

주요업무

We are seeking an experienced AI Model Compiler Engineer to develop and optimize model compilation pipelines for deep learning frameworks. You will work on converting AI models (e.g., ONNX, TensorFlow, PyTorch) into efficient, hardware-optimized code for edge and cloud-based AI processors. Your role will involve compiler optimizations, graph transformations, and hardware-specific acceleration techniques. • Model Compilation Pipeline: Design and implement compilers that translate AI models (ONNX, TensorFlow, PyTorch, etc.) into executable formats for AI accelerators and edge devices. • Graph Optimization: Apply operator fusion, pruning, quantization, and memory optimizations to improve model performance. • Hardware Acceleration: Optimize AI model execution on CPU, GPU, DSP, TPU, or custom AI chips (e.g., NPU, FPGA). • Intermediate Representations (IRs): Work with MLIR, TVM, XLA, Glow, or custom IRs for model transformation. • Performance Tuning: Profile and analyze models using LLVM, Halide, CUDA, OpenCL, or Metal. • Kernel Optimization: Develop low-level math libraries (SIMD, vectorized ops, matrix multiplications, tensor ops) for efficient AI inference. • Custom Operator Support: Implement new AI operators and optimize execution on target hardware. • Cross-Platform Deployment: Enable model portability across multiple architectures and backends. • AI/ML Framework Integration: Extend compiler functionality for PyTorch, TensorFlow, ONNX Runtime, and other ML frameworks. • Debugging & Benchmarking

자격요건

• Education: Bachelor's, Master's, or Ph.D. in Computer Science, Electrical Engineering, or related fields. • Experience: 2+ years in model compilation, AI frameworks, or deep learning accelerators. • Programming Languages: C, C++, Python, and LLVM IR or MLIR. • Compiler Development: Experience with LLVM, TVM, XLA, Halide, Glow, or custom ML compilers. • Graph Transformations: Knowledge of operator fusion, loop unrolling, constant folding, quantization, and tiling techniques. • Hardware Optimization: Experience with SIMD, CUDA, OpenCL, ROCm, or low-level tensor operations. • AI Frameworks: Hands-on with TensorFlow, PyTorch, ONNX, TensorRT, TFLite, or OpenVINO. • Parallel Computing: Experience with multi-threading, vectorization (SSE/AVX), and heterogeneous computing.

우대사항

• Neural Network Compression: Quantization-aware training (QAT), weight pruning, distillation. • Cloud & Edge AI: Deploying models on AWS Inferentia, NVIDIA Jetson, Intel Movidius, or Qualcomm AI chips. • Formal Methods & Verification: Model validation, correctness proofs, and fuzz testing for compiler robustness.

혜택 및 복지

• 나에게 맞는 랩탑은 내가 고른다! 최적의 근무 환경을 만들어 드려요. • 업무 역량 향상을 위해 교육비를 무제한 지원합니다. • 분 단위로 운영되는 10~17시 코어타임제로 출근 시간을 여유롭게! • 맛있는 점심 식사를 제공합니다. 늦게까지 열일할 땐 더 맛있는 저녁 식사도 책임집니다. • 먹고 싶은 간식과 음료를 무한 제공해 드려요. • 생일 축하합니다! 생일을 맞은 구성원에게 1일의 생일 휴가를 드립니다. • 원활한 소통은 모든 업무의 기본 ─ 페블스퀘어가 휴대폰 이용료를 보조해 드립니다. • 가벼운 마음으로 맛있게 먹으며 즐기는 점심 회식으로 기분 전환! • 워라밸도 놓치지 않을 거예요! 사유를 묻지 않는 연차/반차/반반차 제도를 운영합니다. • 구성원의 즐거운 휴가를 응원합니다. 페블스퀘어는 근로자 휴가 지원 사업에 참여하고 있어요. • 뭐니뭐니 해도 건강이 최고죠. 검진 대상자분들께 1일의 건강검진 휴가를 드립니다. • 페블스퀘어의 진심을 담은 경조사 지원금으로 구성원의 기쁨과 슬픔을 함께합니다. • 최고의 복지는 훌륭한 동료! 함께하면 힘이 나는 동료를 추천해 주신 분께 인재 추천 보상금을 드립니다. • 학업에 대한 욕심도 놓지 마세요. 페블스퀘어는 과기부의 석박사 양성사업 참여 기업입니다.