Senior Staff Machine Learning Engineer - Frameworks

Design, build, and optimize ML deployment pipelines and inference frameworks.

• •d-Matrix is developing hardware and software platforms to enhance the efficiency and scalability of generative AI applications.
• •The Model Factory team focuses on building, optimizing, and deploying large-scale machine learning models with an emphasis on efficiency, automation, and scalability for d-Matrix hardware.
• •Key Responsibilities Design, build, and optimize machine learning deployment pipelines for large-scale models.
• •Implement and enhance model inference frameworks.
• •Develop automated workflows for model development, experimentation, and deployment.
• •Collaborate with research, architecture, and engineering teams to improve model performance and efficiency.
• •Requirements BS in Computer Science with 7+ years of strong programming skills in Python and experience with ML frameworks like PyTorch, TensorFlow, or JAX.
• •Hands-on experience with model optimization, quantization, and inference acceleration.
• •Deep understanding of transformer architectures, attention mechanisms, and distributed inference.