3 days ago
Shanghai, ChinaSenior / Staff+
Responsibilities
- Design, train, fine-tune, and evaluate machine-learning and neural-graphics models for gaming workloads and optimize them for Arm CPU/GPU deployment.
- Build data and model-training workflows covering dataset preparation, augmentation, experiment tracking, benchmarking, validation, versioning, automated evaluation, and model packaging or deployment.
- Prototype and integrate neural rendering, super-resolution, denoising, and ray-tracing enhancements into real-time rendering and game-engine pipelines.
- Profile data processing, training, inference, and rendering workloads and improve runtime integration, model precision, GPU execution, and memory efficiency.
- Collaborate with Arm teams and game or engine partners to bring ML and graphics solutions from prototype to deployment in games.
Requirements
- A BS or MS degree in Computer Science, Computer Engineering, Artificial Intelligence, Graphics, or a related field, or equivalent experience.
- Strong programming skills in C/C++ and Python with experience writing maintainable software.
- Hands-on experience developing, training, and evaluating deep-learning models using PyTorch, TensorFlow, or a comparable framework.
- Understanding of model-training fundamentals including data preparation, model selection, loss functions, optimization, evaluation metrics, and reproducible experiments.
- Experience building end-to-end AI/ML workflows involving dataset and model versioning, training or fine-tuning, experiment tracking, automated evaluation, reproducible pipelines, and model deployment.
- Strong understanding of computer graphics and real-time rendering, including graphics pipelines, shaders, and GPU-based workloads.
- In-depth experience with Unreal Engine or Unity, including runtime-code or graphics-feature integration.
- Experience with Android/mobile software development and a modern graphics API such as Vulkan, DirectX 12, or Metal.
- Strong debugging and profiling skills, understanding of Arm Mali or other GPU architectures, and clear written and verbal communication in English and Chinese.
- Preferred qualifications include game-studio integration experience, commercial mobile-game development, edge/mobile ML inference, model conversion, quantization, pruning, runtime acceleration on Arm platforms, and familiarity with Agile/Scrum.
Benefits
- Training and development in a friendly, technology-focused team.
- Opportunities to collaborate with colleagues and participate in social and volunteering activities.
- The role is based at Arm’s Shanghai offices and follows a team-determined hybrid working approach, with details provided during application.
Categories
GamingML Engineering
About Arm
Arm’s foundational technology is defining the future of computing. A future built by the greatest technology ecosystem in the world. A future built on Arm. Arm is everywhere technology matters. Technology matters everywhere. Together, we’ll power every technology revolution moving forward, including cloud computing, automotive and autonomous systems, IoT, the metaverse, and beyond. Changing the world. Again. On Arm.