AI Engineer, AI & Applications

Benefits

Parental leave

Not verified

Non-birth-parent leave

Not verified

Family-building benefits

• Fertility benefits: Not verified
• Adoption assistance: Not verified
• Surrogacy assistance: Not verified

Mental health support

Not verified

Relocation assistance

Not verified

Childcare support

Not verified

Learning budget

Not verified

Verification

Not verified

Salary

Not verified

Was this benefit information wrong? Tell us.

Schedule

Shift type

Not verified

Weekend work

Not verified

Application

Cover letter

Not verified

Assessment

Not verified

Deadline

Not stated

Where they hire

State eligibility is not yet verified.

About this role

AI Engineer, AI & Applications Singapore or Australia Role Summary The AI Engineer will establish Firmus AI Factory as the foundation for efficient, production-grade distributed training by delivering pre-built training recipes (TorchTitan, Megatron etc.), evaluation benchmarks, and model guidance. You'll work with customers and internal teams to optimize training efficiency, define baselines, and document best practices. Your templates and benchmarks are the anchor point for our hyperscale customers' training workflows and our model arena differentiator. Key Responsibilities - Build production-ready training recipes using TorchTitan and Megatron-LM: model configs, parallelism strategies (FSDP, tensor/pipeline parallelism), checkpointing patterns. - Document parameter tuning for different scales (e.g., "to train Llama 7B on 8xH100s, use this config and expect X throughput"). - Create and validate multi-node NCCL communication patterns on AI Factory K8s/Slurm clusters. - Design and build benchmarking suites: accuracy, latency, throughput (tokens/sec), cost per token, energy efficiency, MFU. - Implement offline evaluation harnesses for standardized model comparison and leaderboard tracking. - Conduct fine-tuning experiments (LoRA, QLoRA) where they improve product outcomes (e.g., ops domain data), document gains. - Create training efficiency playbooks and publish benchmark results so customers can optimize workloads. - Partner with job scheduling and orchestration engineers on template integration and other AI engineers and software engineers on model optimization trade-offs for inferencing and AI applications. Skills & Experience - 5-7 years of experience in distributed machine learning (PyTorch/JAX, FSDP, DeepSpeed, multi-node training at 10+ GPUs). - Expert-level understanding of GPU optimization: utilization, memory patterns, communication bottlenecks (NCCL collectives). - Hands-on

Read the full description at job-boards.greenhouse.io. FewerJobs shows a source-linked preview and links to the original posting.