AI/ML Hardware Driver/SDK Software Engineer

Auradine - Santa Clara, CA

Full-time - Entry Level

Santa Clara, CA

About the position

The AI/ML Hardware Driver/SDK Software Engineer at Auradine will be instrumental in designing and developing high-performance device drivers and SDKs for AI/ML infrastructure solutions. This role involves collaborating with cross-functional teams to optimize AI workflows and ensure the reliability and performance of software components in production environments. The position offers an opportunity to work in a dynamic startup environment focused on innovation and the future of AI technology.

Responsibilities

• Design and build scalable, high-performance device drivers, SDKs, and microcodes for AI/ML infrastructure solutions.
• Develop software models of custom hardware to verify functionality and test SDKs and APIs for AI/ML applications.
• Customize and extend open-source AI/ML stacks to optimize performance for Auradine's solutions.
• Collaborate with internal and external teams, including software and hardware engineers and AI researchers, to refine AI/ML solutions.
• Take ownership of code quality, performance, and reliability, ensuring clean, maintainable code and rigorous testing.

Requirements

• Expertise in C/C++ or a similar high-performance language with strong experience in computer algorithms and data structures.
• 2+ years of embedded software development experience, including testing, debugging, and ensuring reliability in production environments.
• Experience with developing user-space and kernel device drivers, with deep knowledge of the Linux kernel and systems programming.
• Experience with developing performance-optimized SDKs, SAI, and APIs in compute or network areas.
• Experience integrating multiple layers of software stacks to utilize hardware capabilities.

Nice-to-haves

• Experience with C-Language ASIC modeling and writing C/SystemC/Simulink models for high-performance silicon.
• Familiarity with software stacks used in AI/ML/HPC systems such as RDMA, RoCE, GPUDirect, CUDA, and Infiniband.
• Expertise in profiling and benchmarking embedded software and hardware systems.

Benefits

• Competitive salary and equity options.
• Flexible working hours and remote work opportunities.
• Health, dental, and vision insurance coverage.
• Professional development and learning opportunities.