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.
