Note: The job is a remote job and is open to candidates in USA. Vedasoft Inc is seeking a Lead Machine Learning Engineer specializing in PyTorch and performance optimization. The role focuses on developing high-quality software, optimizing CPU/GPU execution through systematic performance engineering, and leading technical alignment and mentorship across functions.

Responsibilities

• Ability to lead through influence: mentor others, raise the engineering bar, and drive alignment across functions

Skills

• Deep software engineering experience with expert Python
• Strong hands-on PyTorch experience, including performance-oriented work (e.g., profiler usage; understanding tensor layouts/ops behavior)
• Demonstrated performance engineering discipline: a repeatable loop of profiling → hypothesis → change → before/after evidence
• Solid understanding of how code maps to CPU/GPU execution (parallelism, memory bandwidth/latency, data movement)
• Experience owning improvements in a large, long-lived codebase where correctness and maintainability matter (not just prototypes)
• Ability to lead through influence: mentor others, raise the engineering bar, and drive alignment across functions
• Strategic thinker and independent problem solver—comfortable with ambiguity and high-stakes problem solving
• Proven ability to communicate complex technical topics clearly to non-technical audiences and stakeholders
• Detail-oriented with a commitment to accuracy and quality
• Familiarity with actuarial projections; annuities and quality
• Experience with Monte Carlo/ numerical computing (finance / actuarial is a plus)

Company Overview

Company H1B Sponsorship