AI SDK & Framework Intern （领跑者计划）

领跑者计划是乐鑫为 2027 届海内外学生打造的转正储备实习项目，工作地点设在中国上海。 The Opportunity You will make our high-performance AI silicon accessible to the world. While the Runtime team builds the engine, you build the steering wheel. You will develop the Python SDK, integrating our C++ runtime into ecosystems like PyTorch, ONNX, or IREE. You define how data scientists interact with our chip—from "import aisoc" to running real-world LLMs and vision models end-to-end on our device. Key Responsibilities · Python/C++ Bridging: Build efficient bindings (using pybind11) that allow Python users to drive our low-level C++ Runtime and memory allocator with minimal overhead and zero-copy where possible. · Model Ingestion: Build practical model import tools: weight packing/layout transforms, graph partitioning to supported ops, and integration with existing quantization/calibration workflows. · Developer Experience (DX): Ensure that when a user makes a mistake, they get a helpful Python exception, not a silent segmentation fault. · Golden Reference Examples: Build and maintain the "Hello World" and "Chatbot" demos that verify the entire hardware/software stack is functioning correctly. What We Will Teach You · The internals of modern ML frameworks (how PyTorch dispatch works, how ONNX graphs are structured). · How to build and ship Python wheels and native extensions for our target runtime environment (Embedded Linux). · Techniques for zero-copy memory sharing between Python (numpy) and hardware accelerators.

Must-Have Qualifications · Strong proficiency in Python (you understand decorators, context managers, and the Global Interpreter Lock). · Working knowledge of C++ (you can read a header file and understand what needs to be exposed to Python). · Familiarity with ML Data Structures: You know that a "Tensor" is just a pointer to memory with shape and stride metadata. · Minimum 3 months, at least 4 days per week Nice-to-Have (We Value Projects!) · Experience with ONNX Runtime, TVM, or MLIR. · Experience building Python wheels or C-extensions (pybind11, Cython).