Senior Software Engineer

Job Summary

A company is looking for a Senior Software Engineer, JAX.

Key Responsibilities

• Contribute to the development of JAX and design core components for peak performance
• Collaborate with AI researchers to build future-proof models and tools for AI-based systems
• Bridge the gap between numerical computing, simulation, and deep learning applications in real-world products

Required Qualifications

• BS in Computer Science, Computer Engineering, or related field (or equivalent experience)
• 6+ years of relevant experience
• Proficiency in C/C++ and Python programming
• Experience with machine learning frameworks (e.g., PyTorch, TensorFlow)
• Strong technical foundation in CPU/GPU architectures and modular software design