Machine Learning Software Engineer, PyTorch Specialist



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 is developing infrastructure and tools for automating the process of adapting our unique hardware architecture to run machine learning model training and inference, as well as AI applications.

We are seeking a talented machine learning (ML) PyTorch expert to join our AI Infrastructure team in Belgrade. In this high-visibility, hands-on role, you will be building the AI compiler backend for NextSilicon’s next generation platform.


  • Bachelor's degree in either computer science, computer engineering, or another relevant technical field, or equivalent practical experience.
  • 6+ years of experience in software engineering or a relevant field, or 4+ years of experience if you have a PhD.
  • 3+ years of specialization experience in at least one of the following ML or deep learning domains:
  • ML systems: AI infrastructure, ML accelerators, high performance computing, GPU and/or CPU architecture.
  • ML tools: ML compilers, ML frameworks.
  • ML theory: large scale ML, LLM, ML robustness experience.
  • Hands-on experience applying techniques for splitting and parallelizing model training and inference onto a multiple GPU environment.
  • Experience developing state-of-the-art neural network architectures: 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.

NextSilicon is proud to be an Equal Opportunity Employer. We do not discriminate based upon race, religion, color, age, national origin, gender (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity, gender expression, genetic information, status as a protected veteran, status as an individual with physical or mental disability, or other applicable legally protected characteristics. This policy applies to all employment practices within our organization, including hiring, recruiting, promotion, termination, layoff, recall, leave of absence, compensation, benefits, training, and apprenticeship. NextSilicon makes hiring decisions based solely on qualifications, merit, and business needs at the time.

Send us your CV
Good luck!

For any questions please ask us at