Pensees Systems Pte. Ltd. – Intern for AI Software Engineer

Pensees R&D team is searching for an outstanding student 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 person 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.
• Assist and 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
• Assist refining software designs and implementation strategies with peers
• Improving processes and infrastructure to accelerate our development and contribute to the success of the company

Requirements:

• Any students in Computer Science, Electrical Engineering, Artificial Intelligence or related field with project experience
• Good 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
• 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 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

Benefits:

• Gain hands-on experience in a leading AI company at the forefront of technological advancements especially in Gen AI and its applications.
• Work closely with a diverse team of experts and professionals in the AI industry.
• Opportunity to showcase your skills and potentially grow stronger with the company.
• Flexible working hours to accommodate your schedule, with annual leaves and medical leaves given.
• Potential for performance-based incentives and awards.