Lead Software Engineer
Lead Software Engineer
Employer Active
Job Alert
You will be updated with latest job alerts via email
Share
Job Alert
You will be updated with latest job alerts via email
Not Disclosed
Salary Not Disclosed
1 Vacancy
Job Description
- Profile and optimize ML workloads using tools like cProfile and memoryprofiler 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.
Qualifications :
- 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.
Additional Information :
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 well explore working together!
Remote Work :
Yes
Employment Type :
Full-time
Employment Type
Remote
Company Industry
Key Skills
- Spring
- .NET
- C/C++
- Go
- React
- OOP
- C#
- Data Structures
- JavaScript
- Software Development
- Java
- Distributed Systems
