AI Research Engineer

ABOUT CLIENT

Our client is a leading research company specializing in technology innovation

JOB DESCRIPTION

Focus on improving model usability for users
Conduct training runs and AI experiments
Analyze results and make necessary changes
Collaborate with the product engineering team to implement improvements
Implement and improve upon recent RL techniques like GRPO, DPO, RePO, etc.
Create and manage adaptable, expandable training codebases
Establish and maintain efficient data pipelines, including both synthetic and real data
Ensure training jobs are capable of scaling across multiple GPUs and nodes, such as FSDP, DDP, NCCL
Maintain code health over the long term by writing clean, testable, and reproducible code
Contribute to the enhancement of open source dependencies
(Optional) Publish papers and present research findings

JOB REQUIREMENT

Comprehensive skills in Python and coding using frameworks like PyTorch, or similar
Demonstrated proficiency in training deep learning and reinforcement learning models in practical scenarios
Experience in handling and analyzing large datasets and intricate workflows
Adept understanding of training dynamics, identifying issues and troubleshooting them
Proficiency in job launchers, logging tools (such as Weights & Biases, TensorBoard), and checkpointing systems
An approach of applying engineering precision to research through writing clear code, meticulous design, and reproducible results
Knowledge of TorchScript, ONNX, or custom inference runtimes
Contribution to open-source projects related to PyTorch or machine learning tools
Background in working with transformer models, diffusion models, VLMs, or extensive vision/NLP tasks
Familiarity with batch schedulers (SLURM), cluster environments, and GPU resource management
Ability to collaborate closely with systems engineers or MLOps teams for seamless integration

WHAT'S ON OFFER

Join an exceptional research team to work on significant and impactful projects
Take charge of and influence the primary training code infrastructure utilized by the team
Engage with actual models, real data, and substantial scale challenges, not small-scale problems
Contribute to bridging the gap between research speed and engineering excellence
Enjoy a flexible work setting with a culture that treasures depth, transparency, and inquisitiveness

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:

OSS Software Engineer

Ho Chi Minh - Viet Nam

Product

  • C/C++

Collaborate with a small team of engineers to: Develop software primarily using C language Perform thorough software testing Deliver projects in phased releases Offer customer support for live operations Coordinate software development and support activities with the manager and the core development team based in California, USA

Negotiation

View details

Backend Engineer (Python/Kotlin)

Ho Chi Minh, Ha Noi - Viet Nam

Outsource

  • Python
  • Kotlin
  • AWS

Create and develop financial products on top of a core banking platform called Thought Machine Vault Create and develop event-driven microservices to enhance the functionality of the core banking platform Maintain and enhance service reliability through effective simulation, end-to-end, and performance tests Enhance SRE processes and offer production support for services Write high-quality, sustainable code using TDD Utilize Kubernetes and Docker for scheduling and running microservices Technology stack comprises mainly of Python and Kotlin/Java, with the flexibility to use the most suitable language for problem-solving Utilize PostgreSQL, Aurora, and S3 for persistence Harness the elastic AWS infrastructure Practice continuous integration and delivery Follow the approach of "you build it, you run it."

Negotiation

View details

Senior Full-stack Java Software Engineer

Ho Chi Minh - Viet Nam

Outsource

  • Java
  • Angular
  • Cloud

Create and uphold full-stack web applications using Java as the primary language, Spring Boot for the backend, and Angular (TypeScript) for the frontend. Develop and execute RESTful APIs with a focus on scalability, security, and performance. Engage in system design, technical discussions, and code review. Utilize AWS cloud services for the deployment and management of applications in production. Implement unit tests and integration tests to guarantee code quality. Contribute to Agile/Scrum teams alongside Product Owner, QA, and DevOps. Provide guidance and assistance to less experienced developers.

Negotiation

View details