发现心仪岗位

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.

Nice-to-Have (We Value Projects!)

·       Experience with ONNX Runtime, TVM, or MLIR.

·       Experience building Python wheels or C-extensions (pybind11, Cython).