Machine Learning Architect

Job Summary

To support the development of advanced AI systems, the remote Machine Learning Architect will take hands-on technical ownership of large-scale model architecture, distributed computing, and GPU performance, ensuring reliable operation of modern AI at scale.

Key responsibilities

• Define and evolve production ML architecture across the complete machine learning lifecycle
• Engineer scalable distributed training and inference environments utilizing GPU infrastructure
• Lead optimization efforts focused on GPU performance, including memory utilization and throughput

Required qualifications

• Advanced knowledge of AI model development with practical experience in building complex neural network solutions
• Strong background in scalable machine learning platforms and parallel computing tools
• Demonstrated expertise in GPU performance optimization techniques
• Proven ability to manage architecture across data, training, evaluation, and production operations
• Strong command of modern ML frameworks such as PyTorch or JAX