**
Lightricks is seeking a Large Scale Training Engineer to enhance their internal AI model training framework. The ideal candidate will have strong engineering skills, experience with machine learning pipelines, and a passion for AI innovation.
**
Job Summary
Enhance training throughput of internal framework and enable researchers to pioneer new model concepts.
Profile and optimize the training process, develop high-performance kernels, and integrate advanced techniques.
Design, maintain, and evolve a high-quality, shared codebase emphasizing correctness, readability, extensibility, and maintainability.
Matching Summary
Match Score: 75
**
Lightricks is seeking a Large Scale Training Engineer to enhance their internal AI model training framework. The ideal candidate will have strong engineering skills, experience with machine learning pipelines, and a passion for AI innovation.
**
Skills & Requirements
Must-have
Python and C++ proficiency
GPU/CPU/TPU architecture understanding
Performance optimization for ML
Distributed systems knowledge
Robust machine learning code
Nice-to-have
JAX/Pallas, Triton, CUDA, OpenCL
Kubernetes environments
Key Requirements
Industry experience with ML experiments
Multi-modal ML pipelines experience
Deep understanding of AI accelerator architectures
Experience improving performance without compromising code quality
Passion for driving ML large-scale training workloads