Senior Machine Learning Engineer, On-Device Optimization
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
This role focuses on advancing on-device AI optimization by applying model compression and efficient representation techniques.
You will research and implement quantization, pruning, low-rank factorization, and distillation to reduce model size and improve speed.
Develop deployment methods for state-of-the-art transformer and vision models on-device under hardware constraints.
Lead hardware-aware training strategies to optimize latency, throughput, and memory usage while maintaining accuracy.
Collaborate with software engineers to integrate optimized models into AI companion apps and evaluate cross-framework performance.
Communicate findings through rigorous benchmarking and contribute to open-source ML optimization efforts when possible.
You will research and implement quantization, pruning, low-rank factorization, and distillation to reduce model size and improve speed.
Develop deployment methods for state-of-the-art transformer and vision models on-device under hardware constraints.
Lead hardware-aware training strategies to optimize latency, throughput, and memory usage while maintaining accuracy.
Collaborate with software engineers to integrate optimized models into AI companion apps and evaluate cross-framework performance.
Communicate findings through rigorous benchmarking and contribute to open-source ML optimization efforts when possible.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar openings
Other roles that could suit you.
Remote workpartial
CityMultiple locations (US), US