Staff Software Engineer, Machine Learning, TPU Workload Optimization
Tasks
- Architect training and inference onboarding frameworks
- Collaborate with ML research and performance teams
- Implement single host and multi host inference solutions
- Lead ML onboarding for training and serving
- Optimize training and inference workloads on TPU
- Prepare and optimize large reference models
- Ship stable training and inference stack
- Work with customers to onboard workloads to production
Perks/Benefits
- N/A
Skills/Tech-stack
Data Processing | DeepSpeed | Distributed Computing | Fine Tuning | GPU | HPC | Inference Optimization | Large Scale Training | Large-scale | Latency optimization | Machine Learning | Megatron-LM | Model Debugging | Model Deployment | Model Evaluation | NVIDIA CUDA | PyTorch | Python | SGLang | TPU | TensorFlow | TensorRT-LLM | Throughput Optimization | VLLM
Education
Related jobs
-
Algorithms | C++ | Computer Vision | Data Structures | Deep learningEntry-level InternshipSingapore3h ago
-
Agent | C plus plus | Computer Vision | Deep learning | Language ModelsEntry-level Full TimeSingapore3h ago
-
AI Customer Engineer, Google Cloud SGD 147K-220KAI Agent Development Kit | API Gateway | Agent Development Kit | Agentic AI | As-a-ServiceSenior-level Full TimeSingapore4h ago
-
Customer Engineer, AI, Google Cloud SGD 140K-214KAI Agents | Agentic AI | As-a-Service | Audit Logging | Cloud NetworkingSenior-level Full TimeSingapore4h ago
-
Senior-level Full TimeLand - 249 Jalan Boon Lay, …16h ago
-
Senior Manager (AI Engineer) (Future Capabilities) SGD 110K-170KAgentic AI | Agile | CrewAI | Data Governance | Deep learningSenior-level Full TimeBCA-JEM, Singapore16h ago
-
Senior Data/Data Scientist (Contract) SGD 146K-191KAgentic AI | Agile | Cloud Computing | Data Wrangling | Generative AIContract role | Further assessment | Medical declarationSenior-level Contract Full TimeMAS: MAS Building, Singapore16h ago
-
Actuator integration | BACnet | Block programming | Building Management | Building Management SystemsSenior-level Full TimeNgee Ann Polytechnic, Clementi Campus, Singapore16h ago
-
Agentic AI | Automation | CI/CD | Change Management | DebuggingSenior-level ContractSingapore1d ago
-
A2A | API Integration | Agent systems | Agentic RAG | CI/CDSenior-level ContractSingapore1d ago
-
.NET | AKS | App Service | Azure App | Azure App ServiceMid-level Full TimeSingapore, Singapore1d ago
-
Algorithms | Data Mining | Data Structures | Deep learning | Machine LearningMid-level Full TimeSingapore1d ago
-
Algorithms | Classification | Data Mining | Data Structures | Deep learningMid-level Full TimeSingapore1d ago
-
Apache Flink | Apache HBase | Apache Kafka | Apache Spark | C#Senior-level Full TimeSingapore1d ago
-
Control Theory | Domain Randomization | Imitation Learning | Isaac Lab | Isaac-GymFast iteration feedback loops | High visibility projects | Open source contributionsMid-level Full TimeSingapore1d ago
-
Quantitative Developer SGD 140K-180KBash | C++ | NumPy | Pandas | PolarsContinuous learning | Free meals | Hybrid working | Paid time off | Social eventsMid-level Full TimeSingapore R1d ago
-
A I | A I Inference | A/B | A/B Testing | Agent systemsDiversity and inclusion | Inclusive work environment | Open workspaces | Training and mentoringMid-level Full TimeSingapore, SG / Austin, TX1d ago
-
[Lta-Itcd] Senior Ai Engineer / Ai Engineer SGD 74K-100KAI Governance | Agentic AI | Cloud Platforms | Computer Vision | Data ManagementMedical assessmentSenior-level Full TimeLTA HSO B6 02, Singapore1d ago
-
Full Stack AI Engineer (Contract) SGD 90K-146KAngular | Audit Logging | Authentication | Authorization | CachingMid-level Contract Full TimeMAS: MAS Building, Singapore1d ago
-
C++ | CAN bus | Clean architecture | Code review | Computer VisionSenior-level Full TimeRepublic Polytechnic, Singapore1d ago
-
Data Engineer SGD 136K-200KAWS Database | AWS Database Migration Service | AWS Glue | AWS Lambda | AWS S3Access to certifications coaching and learning opportunities | Be Well programs | Continuous feedback | Flexible work environment | Hybrid workMid-level Full TimeKSG51588 Singapore (KSG51588) R1d ago
-
M10 - Data Engineer SGD 60K-60KAPI Gateway | AWS CloudFormation | AWS Fargate | AWS IAM | AWS LambdaMid-level Full TimeSingapore2d ago
-
Senior Software Engineer (Data), Ai & Data Platforms SGD 149K-205KAPI | AWS | Algorithms | Anomaly Detection | Apache SparkSenior-level Full TimeSingapore2d ago
-
Amazon Web Services | Data Pipelines | Data Quality | Data Storage | Data ValidationAccess to cutting-edge technology | Career advancement | Flexible working hours | Health insurance | Holiday scheduleMid-level Full TimeSingapore, Singapore, Singapore R2d ago
-
AWS | Apache Spark | Azure | Google Cloud | HadoopAccess to cutting-edge technology | Career advancement | Collaborative work environment | Dental insurance | Flexible working hoursMid-level Full TimeSingapore, Singapore, Singapore R2d ago