Software Engineer

Collaborate with engineers to design, develop, and test inference solutions for state-of-the-art AI models.

Implement, optimize, and evaluate AI models using Python, C++, CUDA, and NCCL (previous exposure helpful - deep expertise not required).

Monitor and maintain production model-serving systems.

Work on new features, fix bugs, and contribute to code reviews.

Participate in daily standups, design reviews, and team discussions.

Explore new AI/ML techniques and tools, and experiment with improving model performance.

Try new things. Ship stuff.

Bachelor’s or Master’s degree in Computer Science, Computer Engineering, or a related field (completed or in final year).

Strong fundamentals in data structures, algorithms, and software design.

Proficiency in Python, including experience with AI/ML libraries and frameworks (e.g., NumPy, pandas, SciPy, TensorFlow, PyTorch).

Experience with AI/ML through coursework, research, personal projects, full-time employment, or internships.

Familiarity with AI models, Transformers and Diffusers.

Experience with version control systems (e.g., Git) and agile development methodologies.

Excellent problem-solving skills, with the ability to debug and optimize code.

Strong communication and collaboration skills.

Curiosity, willingness to learn, and desire to build real systems.

Exposure to C++, CUDA, or AI inference.