Semesteria Logo
Firmus Technologies

AI Engineer

Firmus Technologies

Posted

3 weeks ago

Singapore

Onsite

SGD 10K

Mid Level

Full Time

Match

Skills

Experience

Industry

Sema

Sema Summary

The AI Engineer will establish efficient, production-grade distributed training at Firmus Technologies. This role involves optimizing training efficiency and documenting best practices for AI workflows.

About Company

Firmus Technologies creates the most efficient AI infrastructure, focusing on innovative engineering solutions for sustainable AI practices.

Core Requirements

  • 5-7 years of experience in distributed machine learning.
  • Expert-level understanding of GPU optimization.
  • Hands-on experience with PyTorch or JAX.
  • Strong benchmarking methodology.
  • Familiarity with production training frameworks.

Responsibilities

  • Build production-ready training recipes using TorchTitan and Megatron-LM.
  • Document parameter tuning for different scales.
  • Create and validate multi-node NCCL communication patterns.
  • Design and build benchmarking suites for AI models.
  • Implement offline evaluation harnesses for model comparison.
  • Conduct fine-tuning experiments to improve product outcomes.
  • Create training efficiency playbooks and publish benchmark results.

Must Have skills

Distributed machine learningGPU optimizationBenchmarking methodologyDocumentation skillsMentorship abilities

Job Keywords

AI EngineerMachine LearningGPU OptimizationBenchmarkingDistributed Systems

Similar Jobs