Senior AI Engineer
Role details
Job location
Tech stack
Job description
We are looking for an outstanding individual with expertise in training large-scale AI models, including Large Language Models (LLMs) or equivalent architectures. This person will play a key role in the development and optimization of AI workloads, collaborating closely with hardware and software engineers in a co-design environment., · Design, train, and optimize large-scale AI models, including LLMs and similar architectures.
· Collaborate with hardware and software engineers to ensure efficient model deployment on custom silicon solutions.
· Develop new techniques for improving training efficiency, model accuracy, and hardware utilization.
· Implement distributed training strategies for large-scale models across high-performance computing infrastructures.
· Strong programming skills in Python, (Go, C++,…is considered a plus)
· Explore and implement advanced AI/ML techniques, such as sparsity, quantization, and knowledge distillation.
· Work with state-of-the-art frameworks such as TensorFlow, PyTorch, JAX, and Hugging Face Transformers.
· Conduct performance benchmarking and profiling of ML workloads on custom accelerators.
· Stay up to date with advancements in deep learning and AI hardware to bring innovative ideas into Openchip's AI strategy.
· Participate in technical discussions and contribute to Openchip's AI research and development roadmap.
Requirements
· BSc, MSc or PhD in Computer Science, Machine Learning, Artificial Intelligence, or a related field.
· Proven experience in training and deploying large AI models, particularly LLMs or equivalent large architectures.
· Strong programming skills in Python and proficiency with ML frameworks such as TensorFlow or PyTorch.
· Experience with distributed training, model parallelism, and large-scale data handling.
· Strong knowledge of deep learning model optimization techniques.
· Experience working in a hardware/software co-design environment is highly desirable.
· Fluent in English is a must; knowledge of additional European languages (German, Dutch, Spanish, French, or Italian) is a plus.
Soft Skills:
· The candidate should be equipped with a unique skills-set: Self-starter, self-motivated, humility, excellent communications skills; outstanding human qualities as honesty, integrity, fellowship, generosity and commitment with his/her mission to change the world.
· Excellent communication and collaboration skills to work effectively in a multidisciplinary team.
· A strong commitment to innovation and problem-solving, a can-do attitude.
· Passionate about driving the next generation of AI hardware/software integration.
· Willingness to mentor junior engineers and contribute to a collaborative learning culture.