Lead Software Engineer (Machine Learning)

Montevideo , Uruguay
full-time
  • Profile and optimize ML workloads using tools like cProfile and memory_profiler, identifying CPU/GPU bottlenecks. 
  • Apply vectorization and parallelization techniques (GPU/TPU, multi-threading) to accelerate performance. 
  • Work with large-scale datasets efficiently using Pandas, NumPy, and JAX, minimizing costly type conversions. 
  • Implement and optimize ODE solvers and computational workflows. 
  • Leverage JAX best practices, including JIT compilation and maintaining computational graph integrity. 
  • Refactor code for scalability, maintainability, and consistency. 
  • Strong Python programming skills with experience in performance optimization. 
  • Proficiency in profiling tools and vectorized computation. 
  • Solid understanding of parallelization (multi-threading, GPU/TPU). 
  • Experience with high-performance computing in ML contexts. 
  • Familiarity with machine learning workflows and data processing at scale. 
  • Experience working with JAX and its performance optimization features is a nice to have skill. 
  • Experience setting up automated benchmarking and performance regression tests in CI/CD pipelines. 

What about languages?

Advanced English proficiency required.

How much experience must I have?

Minimum of 8 years in software development, with experience with Machine Learning as well.  

📚 Learning Opportunities: 

  • Certifications in AWS (we are AWS Partners), Databricks, and Snowflake. 
  • Access to AI learning paths to stay up to date with the latest technologies. 
  • Study plans, courses, and additional certifications tailored to your role. 
  • Access to Udemy Business, offering thousands of courses to boost your technical and soft skills. 
  • English lessons to support your professional communication. 

👩‍🏫 Mentoring and Development: 

  • Career development plans and mentorship programs to help shape your path. 

🎁 Celebrations & Support: 

  • Special day rewards to celebrate birthdays, work anniversaries, and other personal milestones. 
  • Company-provided equipment.  

⚖️ Flexible working options to help you strike the right balance.    

Other benefits may vary according to your location in LATAM. For detailed information regarding the benefits applicable to your specific location, please consult with one of our recruiters. 

So what are the next steps?  

Our team is eager to learn about you! Send us your resume or LinkedIn profile below and we’ll explore working together! 

 BLEND360 is an award-winning, new breed Data Science Consultancy focused on powering exceptional results for our Fortune 500/1000 clients and other major organizations. We are a growing company—born at the intersection of advanced analytics, data, and technology.Who we are:People are everything here at BLEND360.  We are inspired by advancing our Client’s most critical initiatives, products and projects by matching our clients with the right talent. BLEND360 has been among the Inc. 5000 fastest growing companies 8 years in a row, and we’re very proud of our World Class NPS score. Our success is a direct result of our passion for advancing the careers of the talented people we work with every day. When you work at BLEND360, you will:Collaborate with a smart, passionate group of people who are invested in your success.Partner with an impressive list of clients, who value Blend360’s services and the world class experience we deliver with every engagement. Thrive with a company and leadership team who are committed to growth.

View all jobs
Get hired quicker

Be the first to apply. Receive an email whenever similar jobs are posted.

Ace your job interview

Understand the required skills and qualifications, anticipate the questions you may be asked, and study well-prepared answers using our sample responses.

Machine Learning Engineer Q&A's
Report this job
Apply for this job