5 days ago
Responsibilities
- Implement latest machine learning models and optimize them for performance and accuracy.
- Test and evaluate new internal software releases and provide feedback.
- Benchmark models and key ML techniques to identify performance bottlenecks.
- Design and conduct experiments on novel AI methods and evaluate results.
- Collaborate with Research, Software, and Product teams to build and test AI hardware.
- Engage with the AI community and stay updated on the latest developments.
Requirements
- Bachelor/Master's/PhD in Machine Learning, Computer Science, Maths, Data Science, or related field.
- Proficiency in deep learning frameworks like PyTorch/JAX.
- Strong Python or C++ software development skills.
- Expertise in deep learning from model training to optimization and evaluation.
- Experience in distributed training or inference of ML models across 64+ accelerators.
- Capable of designing, executing, and reporting from ML experiments.
- Developed understanding of performance bottlenecks and solutions.
- Ability to work quickly in a dynamic environment.
- Enjoy cross-functional collaboration.
- Strong communication skills to explain complex concepts.
Benefits
- Flexible working arrangements.
- Generous annual leave policy.
- Private medical insurance and health cash plan.
- Dental plan and pension matched up to 5%.
- Life assurance and income protection.
- Generous parental leave policy.
- Employee assistance programme for health and wellbeing.
- Healthy food and snacks at the office.