AI Research Engineer

JOB DESCRIPTION

Responsibilities
Obsess over model usability for users
Conduct training runs and AI experiments
Scrutinize results and iterate
Continuously productionize checkpoints with the product engineering team
What you'll do
Iterate upon latest RL techniques, e.g. GRPO, DPO, RePO, etc.
Build and maintain modular, scalable training codebases
Develop and maintain efficient data pipelines (synthetic and real)
Ensure training jobs can scale across multiple GPUs and nodes (e.g., FSDP, DDP, NCCL)
Maintain long-term code health: write clean, testable and reproducible code
Contribute to upstream open source dependencies
(Optionally) Publish papers and present findings

JOB REQUIREMENT

Deep expertise in Python and training codebases, e.g. PyTorch, or equivalent
Proven experience training deep learning & reinforcement learning models in real-world settings
Experience working with large datasets, complex pipelines
Understanding of training dynamics: what goes wrong, and how to fix it
Familiarity with job launchers, logging tools (e.g., Weights & Biases, TensorBoard), and checkpointing systems
A mindset of engineering rigor applied to research: readable code, thoughtful design, and reproducibility
Bonus Points
Experience with TorchScript, ONNX, or custom inference runtimes
Open Source contributions to PyTorch or ML tooling
Experience working on transformer models, diffusion models, VLMs, or large-scale vision/NLP tasks
Familiarity with batch schedulers (SLURM), cluster environments, and GPU resource management
Ability to collaborate closely with systems engineers or MLOps teams to ensure smooth integration

WHAT'S ON OFFER

Collaborate with a world-class research team on meaningful, high-impact projects
Own and shape the core training code infrastructure used daily by the team
Work on real models, real data, and real scale - not toy problems
Help bridge the gap between research velocity and engineering quality
Flexible work environment with a culture that values depth, clarity, and curiosity

CONTACT

PEGASI – IT Recruitment Consultancy | Email: recruit@pegasi.com.vn | Tel: +84 28 3622 8666
We are PEGASI – IT Recruitment Consultancy in Vietnam. If you are looking for new opportunity for your career path, kindly visit our website www.pegasi.com.vn for your reference. Thank you!

Job Summary

Company Type:

Product

Technical Skills:

Machine Learning, Python

Location:

Ho Chi Minh - Viet Nam

Working Policy:

Onsite

Salary:

Negotiation

Job ID:

J01951

Status:

Active

Related Job:

Product Quality Engineer

Ho Chi Minh - Viet Nam

Product

  • Automation Test

Design and execute comprehensive test strategies for across web, API platform, desktop and mobile platforms Build automated test suites and Auto QA agents across continuous releases, model updates, and API integrations Own build and CI/CD pipelines, ensuring product runs across all major operating systems: Windows, MacOS and Linux Ensure compatibility between web clients and various server node versions, testing upgrade paths and backwards compatibility Validate resource management and performance optimizations when running models across different hardware configurations, including with GPU acceleration Live in our Discord community and GitHub Issues: transform community feedback into actionable test cases Own release cycles, triage bugs, and sound the alarm Create documentation that actually helps users solve problems

Negotiation

View details

Android Software Engineer (Senior)

Ho Chi Minh - Viet Nam

Product

  • Android
  • Kotlin
  • Java

We are hiring Android Software Engineers (Senior). As an Engineer, you will work on building scalable, high-performance applications with a focus on performance optimization and clean architecture. Implement Android (Kotlin) and Flutter components of our SaaS product Drive the entire mobile software lifecycle, from prototyping to commercialization and post-launch support Build clean, well-tested, and maintainable code that aligns with our cross-platform architecture and performance goals Participate in design reviews, code reviews, and contribute to cross-functional discussions and continuous improvement Partner with Product, Design, and Backend teams to deliver quality Android applications#For Seniors: Propose architecture and design reviews to ensure high standards in software quality Set a high technical standard for the team through code and design reviews Enhance engineering processes, including sprint planning, stand-ups, and long-term planning Mentor junior/middle engineers, foster growth, and technical skill development within the team

Negotiation

View details

iOS Software Engineer (Senior)

Ho Chi Minh - Viet Nam

Product

  • iOS

You will work closely with other teams to design, build new products, platforms and features. Develop and maintain digitalization applications and platform SDKs. Identify and correct bottlenecks and fix bugs. Collaborate with cross-departmental teams and global partners to deliver robust, scalable, and user-focused solutions. Conduct research on emerging technologies to drive innovation and continuous improvement.

Negotiation

View details