A Software Engineer (AI/ML) at PT Lab will work at the intersection of code and physical systems. You will contribute directly to building the MVP of our True Core System (TCS) — the software backbone of our physiotherapy robotics platform — while also helping optimize the data engineering pipelines that power our Embodied AI track. This is a hands-on, high-ownership role for engineers who want to ship real code that runs on real hardware.

You Will :

Design and implement a reinforcement learning training program for our dynamic body weight support physiotherapy systems to enable smart, adaptive patient assistance
Architect and manage high-fidelity data collection pipelines for humanoid teleoperation and behavior cloning.
Optimize ML models for resource-constrained edge devices, ensuring low-latency, real-time inference on hardware.
Collaborate closely with mechanical and software engineers to integrate AI models into our core robotic control stacks.
Establish data-driven testing and validation protocols to evaluate model performance and safety in real-world physical environments.
Take ownership of the AI software architecture, moving rapidly from simulation and high-level planning to hardware deployment.

Requirement:

Basic Qualifications:

Bachelor's degree in Software Engineering, Computer Science, AI/Machine Learning, Robotics, or a related field (or equivalent hands-on experience).
1+ years of hands-on experience developing and deploying machine learning models, computer vision systems, or reinforcement learning algorithms.
Proficiency in Python, C++, and deep learning frameworks (PyTorch, TensorFlow)
Demonstrated experience with Edge AI, including model quantization, optimization, and deployment on embedded computing hardware (e.g., Nvidia Jetson).
Strong understanding of the fundamental mathematics behind machine learning, linear algebra, and data structures.
Portfolio or evidence of practical AI projects interacting with physical systems, sensor data, or complex physics simulations.
Fluency in English.

Preferred Skills and Experience:

Leadership in AI or robotics projects, demonstrating the ability to drive an ML feature from concept to physical deployment.
Background in processing and managing large-scale sensor data streams (kinematic data, load cells, vision) for continuous model training.
Experience deploying reinforcement learning models from simulation (Sim2Real) to physical hardware.
Familiarity with robotics middleware (ROS2) and integrating ML pipelines with hardware control loops.
Clear communication skills to translate complex AI behavior and performance metrics to cross-functional engineering teams.
A hands-on, builder's mindset with a preference for fast, iterative deployment over theoretical perfection.

Special Note:

PT Lab is an innovation lab pushing the boundaries of robotics, embodied AI, and hardware solutions to create value for Vietnam and the world. We're building a brilliant, agile team to develop cutting-edge projects from smart physiotherapy devices to high-fidelity teleoperation systems and next-gen delivery drones. Our work is hands-on, fast-paced, and driven by a mission to turn bold ideas into reality using a first-principles approach. If you thrive in an environment where creativity, ownership, and technical excellence are paramount, PT Lab is the place for you.

Senior AI/ML Engineer

Junior Software Engineer (AI/ML) - PT Lab