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