Senior AI or AI Software Engineer

at  PENSEES SYSTEMS PTE LTD

JOB DESCRIPTION:

Pensees R&D team is searching for an outstanding software engineer to build the AIGC framework of the future. Join us in developing the AIGC platform for supporting both the server and edge devices. We are looking for a dedicated engineer with expertise in GPU programming and performance optimizations. Are you creative, quick to come up with innovative solutions, and eager to apply your skills to empower developers to take advantage of novel technology? If you love the craft of software engineering and relish the chance to impact the future of the industry, this role may be perfect for you.

Key Responsibilities:

• Contribute to the development of AIGC framework by enabling performance optimizations and GPU acceleration within the compiler and runtime.
• Partner with engineers, product managers, and application teams to build and refine tools that accelerate groundbreaking research
• Solve difficult problems at the intersection of compilers, application and edge computing and deliver optimal solutions
• Discuss and refining software designs and implementation strategies with peers
• Improving processes and infrastructure to accelerate our development and contribute to the success of the company

MINIMUM JOB REQUIREMENTS:

• Bachelor degree above in Computer Science, Electrical Engineering, Artificial Intelligence or related field with number of years of working experience
• Proficient in GPU programming either using NVIDIA CUDA libraries such as CUBLAS, CuDNN or NPP, or using Apple low level libraries such as Metal, MPS or MPS Graph
• Proficient in C++, Objective C or Swift programming
• Solid understanding of performance profiling, AI model optimizations, and compiler fundamentals
• Ability to quickly develop expertise in new domains and products, and eagerness to master new challenges.
• Strong communication and collaborative abilities along with analytical and problem, solving skills

PREFERENCE WILL BE TO CANDIDATES WITH THE FOLLOWING QUALIFICATIONS:

• Deep understanding of Apple’s Metal architecture and its usage
• Have experience with low-level development in leveraging hardware resource
• Staying abreast of the latest advancements in deep learning frameworks like TensorFlow, PyTorch or JAX, maximizing their potential on GPUs
• Extensive knowledge about the testing and deployment of infrastructures, and a passion for improving sophisticated engineering systems
• Expertise in parallel programming and multi-processor systems
• Ability to work on large-scale software projects, and a proven track record of building performant and robust production software

• Contribute to the development of AIGC framework by enabling performance optimizations and GPU acceleration within the compiler and runtime.
• Partner with engineers, product managers, and application teams to build and refine tools that accelerate groundbreaking research
• Solve difficult problems at the intersection of compilers, application and edge computing and deliver optimal solutions
• Discuss and refining software designs and implementation strategies with peers
• Improving processes and infrastructure to accelerate our development and contribute to the success of the compan