Lead Software Engineer

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! Blend is a premier AI services provider, committed to co-creating meaningful impact for its clients through the power of data science, AI, technology, and people. With a mission to fuel bold visions, Blend tackles significant challenges by seamlessly aligning human expertise with artificial intelligence. The company is dedicated to unlocking value and fostering innovation for its clients by harnessing world-class people and data-driven strategy. We believe that the power of people and AI can have a meaningful impact on your world, creating more fulfilling work and projects for our people and clients. For more information, visit www.blend360.com What is the position about? We are looking for a highly skilled Software Engineer with expertise in performance optimization to enhance and scale Machine Learning systems for a cutting-edge biotechnology application. The ideal candidate will be passionate about profiling, optimizing, and refactoring computational pipelines to achieve peak efficiency in high-performance computing environments. Originally posted on Himalayas