ML Compute Acceleration Engineer
ML Compute Acceleration Engineer
Posted on :
02-08-2025
Employer Active
1 Vacancy
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Share
Send me jobs like this
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Monthly Salary
USD 147400 - 272100
Vacancy
1 Vacancy
Posted on :
02-08-2025
Job Description
Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating machine learning networks on Apple Silicon using GPU and Neural has the opportunity to influence the design of compute and programming models in next generation GPU and Neural Engine : * Adding optimizations in machine learning computation graph. * Defining and implementing APIs in Metal Performance Shaders Graph investigating new algorithms. * Developing and maintaining MLIR dialect in Apple and open source with upgrades using latest LLVM. * Performing in-depth analysis compiler and kernel level optimizations to ensure the best possible performance across hardware families. * Tune GPU and Neural Engine accelerated compute across products. * Tuning the cost model and optimizing runtime dispatch to multiple IPs to get best performance on Apple deliverables: * GPU Compute acceleration technology. * Apple Intelligence implementation and acceleration. * Optimized compute graphs across this sounds of interest we would love to hear from you!
- Proven programming and problem-solving skills.
- Good understanding of machine learning fundamentals.
- GPU compute programming models & optimization techniques.
- GPU compute framework development maintenance and optimization.
- Experience with system level programming and computer architecture.
- Experience with high performance parallel programming GPU programming or LLVM/MLIR compiler infrastructure is a plus.
- Background in mathematics including linear algebra and numerical methods.
- Strong communication and collaboration skills.
- Strong background of building high performance production quality software on schedule.
- Experience with compiler technologies.
- Experience with adding computational graph support runtime or device backend to Machine learning libraries (TensorFlow PyTorch or JAX) support is a plus.
Employment Type
Full Time
Key Skills
Report This Job
Disclaimer: Drjobpro.com is only a platform that connects job seekers and employers. Applicants are advised to conduct their own independent research into the credentials of the prospective employer.We always make certain that our clients do not endorse any request for money payments, thus we advise against sharing any personal or bank-related information with any third party. If you suspect fraud or malpractice, please contact us via contact us page.
