Serbia- Python Software Engineer

NextSilicon is reimagining high-performance computing. Our accelerated compute solutions leverage intelligent adaptive algorithms to vastly accelerate supercomputers, driving them forward into a new generation. Our new software-defined hardware architecture enables HPC & AI to fulfill its promise of breakthroughs in all fields of advanced research. 

At NextSilicon, everything we do is guided by three core values:

• Professionalism: We strive for exceptional results through professionalism and unwavering dedication to quality and performance. 
• Unity: Collaboration is key to success. That's why we foster a work environment where every employee can feel valued and heard. 
• Impact: We're passionate about developing technologies that make a meaningful impact on industries, communities, and individuals worldwide.

The AI Infrastructure team develops infrastructure and tools that allow our unique hardware architecture to run machine AI models training and inference, as well as other flows.

We are seeking a Python expert to join our AI Infrastructure team in Belgrade. In this high-visibility, hands-on role you will be contributing to the AI compiler backend and the runtime components of NextSilicon’s next generation platform.

• Bachelor's degree in either computer science, computer engineering, or another relevant technical field, or equivalent practical experience.
• 5+ years of experience in Python software engineering, or 4+ years of experience if you have a PhD.
• Experience with ML systems - AI infrastructure, ML accelerators, high performance computing, GPU and/or CPU architecture: an advantage.
• Experience with ML compilers, ML frameworks: an advantage.
• Experience with ML theory - large scale ML, LLM, ML robustness experience: an advantage.
• Experience with open source software development: an advantage.
• Experience developing and upstreaming into the PyTorch open source community: an advantage.

• Develop a state-of-the-art AI framework and compiler stack that will run seamlessly on our next generation hardware, and deliver superior acceleration of AI applications.
• Contribute to the development of an industry leading PyTorch AI framework core compiler. 
• Analyze DL/ML networks and AI applications to understand how they may run more efficiently on the NextSilicon architecture, and develop and implement a compiler backend to support these optimizations. 
• Help to define NextSilicon next generation AI hardware architecture by modeling hardware performance using software simulations.