AI Infrastructure Engineer

Pokee AI is seeking an AI Infrastructure Engineer to build and optimize systems for their RL-trained AI agents. The role involves designing scalable training and inference infrastructure, optimizing model serving, and collaborating with research scientists to productionize algorithms.

Responsibilities:

• Design, build, and maintain scalable training and inference infrastructure for RL-based AI agent models
• Optimize model serving for latency, throughput, and cost across cloud (AWS, GCP) and on-premise/on-device environments
• Develop and manage CI/CD pipelines, experiment tracking, and model versioning systems
• Implement efficient data pipelines for training data collection, preprocessing, and reward signal computation
• Collaborate with research scientists to productionize new algorithms and model architectures
• Ensure infrastructure meets enterprise requirements for reliability, security, and compliance (SOC 2, data residency)