Lead Software Engineer - Remote

Overview

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.

In Short

• Profile and optimize ML workloads using tools like cProfile and memory_profiler.
• Apply vectorization and parallelization techniques to accelerate performance.
• Work with large-scale datasets efficiently using Pandas, NumPy, and JAX.
• Implement and optimize ODE solvers and computational workflows.
• Leverage JAX best practices, including JIT compilation.
• Refactor code for scalability, maintainability, and consistency.

Requirements

• Strong Python programming skills with experience in performance optimization.
• Proficiency in profiling tools and vectorized computation.
• Solid understanding of parallelization techniques.
• Experience with high-performance computing in ML contexts.
• Familiarity with machine learning workflows and data processing at scale.
• Experience with JAX is a plus.
• Experience setting up automated benchmarking in CI/CD pipelines.

Benefits

• Certifications in AWS, Databricks, and Snowflake.
• Access to AI learning paths and Udemy Business courses.
• Career development plans and mentorship programs.
• Special day rewards for personal milestones.
• Company-provided equipment.
• Flexible working options.