Machine Learning Architect

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