
Lead Software Engineer
- Montevideo
- Permanente
- Tiempo completo
- 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.
- 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.
- Special day rewards to celebrate birthdays, work anniversaries, and other personal milestones.
- Company-provided equipment.