about 1 hour ago
Responsibilities
- Implement and optimize machine learning models for performance and accuracy on large-scale systems.
- Test and evaluate new software releases, providing feedback and conducting code reviews.
- Benchmark models to identify performance bottlenecks and improve efficiency.
- Design and conduct experiments on novel AI methods and evaluate results.
- Collaborate with cross-functional teams to define and test next-generation AI hardware.
- Engage with the AI community to stay updated on the latest developments.
Requirements
- Bachelor's, Master's, or PhD in Machine Learning, Computer Science, Maths, Data Science, or related field.
- Proficiency in deep learning frameworks like PyTorch or JAX.
- Strong software development skills in Python or C++.
- Expertise in deep learning from model training to optimization and evaluation.
- Experience in distributed training or inference of ML models across 64+ accelerators.
- Ability to design, execute, and report on ML experiments.
- Strong understanding of performance bottlenecks and solutions.
- Excellent communication skills for explaining 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 and employee assistance programme.
- Healthy food and snacks at the office, including a barista bar.