Senior AI Software Developer - Python

We are seeking a highly skilled and innovative Senior AI Software Developer with deep expertise in Python, Computer Vision (CV), and NVIDIA platforms. In this pivotal role, you will be responsible for designing, developing, and deploying cutting-edge data and AI solutions. You will leverage your experience to build, optimize, and scale intelligent systems that drive significant impact, working with large datasets and ensuring high performance in production environments.

Responsibilities:

• Develop AI/ML Solutions: Design, develop, and implement robust and scalable AI/ML models and applications primarily using Python, with a strong focus on Computer Vision.

• NVIDIA Platform Optimization: Utilize and optimize solutions for NVIDIA hardware and software platforms (e.g., CUDA, TensorRT, DeepStream, Jetson, NVIDIA GPUs) to achieve high performance and efficiency for inference and training.

• Data Pipeline & Processing: Work with large and complex datasets, designing and implementing efficient data pipelines for model training, validation, and inference.

• Model Deployment & MLOps: Lead the deployment of AI/ML models into production, implementing MLOps best practices for monitoring, versioning, and lifecycle management.

• Research & Innovation: Stay abreast of the latest advancements in AI, machine learning, and computer vision, evaluating and integrating new techniques to enhance our solutions.

• Cross-functional Collaboration: Collaborate closely with data scientists, machine learning engineers, product managers, and other stakeholders to translate requirements into technical designs and deliver impactful AI products.

• Performance Tuning: Identify and resolve performance bottlenecks in AI models and underlying infrastructure, ensuring optimal resource utilization.

• 5+ years of professional experience in AI/ML software development, with a strong focus on building and deploying data-driven solutions.

• Exceptional proficiency in Python for AI/ML development.

• Deep expertise in Computer Vision (CV), including experience with various CV algorithms, techniques, and relevant libraries (e.g., OpenCV, scikit-image) and frameworks (e.g., TensorFlow, PyTorch).

• Hands-on experience with NVIDIA technologies for AI acceleration, such as CUDA, GPU programming, TensorRT, and familiarity with platforms like Jetson or DeepStream.

• Proven experience working with large datasets and building robust data pipelines for AI/ML workflows.

• Solid understanding of machine learning principles, model training, evaluation, and deployment strategies.

• Experience with cloud platforms (e.g., AWS, GCP, Azure) for deploying and managing AI/ML workloads.

• Strong problem-solving, debugging, and optimization skills for complex AI systems.

• Excellent communication and collaboration skills, with the ability to articulate technical concepts clearly.

Perks you enjoy at KMS Mexico

• Mexican law benefits
• 15 days of PTO (in year zero, from the first year onwards it is 3 days per year).
• 5 days' leave for the death of immediate family members, negotiable.
• Major Medical Expenses Insurance with coverage for immediate dependents (spouse and children).
• Annual performance bonus (≈10% of annualized salary).
• Annual salary adjustment.
• Employee Referral Bonus.
• Paid Certifications / Courses
• Coursera License.
• 5% Savings Fund.
• 5% Grocery Vouchers.