About this role
The AI Engineer will build production-ready training recipes and benchmarking suites to optimize training efficiency for customers. They will also document best practices and partner with other engineers to enhance model optimization.
Candidates should have 5-7 years of experience in distributed machine learning and a strong understanding of GPU optimization. Familiarity with production training frameworks and benchmarking methodologies is also required.
What they're looking for
Distributed Machine LearningPyTorchJAXFSDPDeepSpeedMulti-node TrainingGPU OptimizationBenchmarking Methodology
Frequently asked questions
What does a AI Engineer, AI & Applications at Firmus Technologies do?
The AI Engineer will build production-ready training recipes and benchmarking suites to optimize training efficiency for customers. They will also document best practices and partner with other engineers to enhance model optimization. Candidates should have 5-7 years of experience in distributed mac…
What skills does this AI Engineer, AI & Applications role need?
Key skills for this role include Distributed Machine Learning, PyTorch, JAX, FSDP, DeepSpeed, Multi-node Training.
How much does a AI Engineer, AI & Applications at Firmus Technologies pay?
The employer did not list a salary for this role. Most similar Singapore roles publish their band on the job page.
Is this AI Engineer, AI & Applications role remote, hybrid, or on-site?
This role is on-site, based in Australia.
How do I apply for this AI Engineer, AI & Applications role?
You can apply directly on Firmus Technologies 's careers page. ApplyLah can tailor your résumé and cover letter to this exact role in seconds first.