Full-Stack Machine Learning Engineer
Location: Paris (Hybrid)
Reports to: CTO
About:
We are working with an AI-driven startup focused on transforming carbon-intensive industries for a more sustainable future. With over €5 million in funding, we are building a state-of-the-art AI platform to discover new materials that reduce CO2 emissions. Join our passionate team and help drive the energy transition.
Key Responsibilities:
- Software Development: Write and maintain high-quality Python code, following best practices such as testing, code reviews, and version control (Git).
- Data Management: Design efficient data storage and retrieval systems for handling large datasets critical to machine learning and chemical research.
- Performance Optimization: Implement parallelized code to optimize CPU-bound tasks, particularly in reinforcement learning or simulation environments.
- Cloud Infrastructure: Build and maintain cloud-based HPC/bare-metal infrastructure, MLOps pipelines, and distributed training workflows.
- ML Engineer Support: Help optimize GPU performance for deep learning tasks and provide debugging support for team members.
- Collaboration: Work closely with interdisciplinary teams of engineers and scientists, ensuring clear communication and documentation.
Requirements:
- Master's degree or PhD in Computer Science, Engineering, or related field with strong software development expertise.
- Expert in Python and experienced with scientific computing, low-level optimizations, and managing hybrid closed/open code bases.
- Knowledge of SQL, NoSQL databases, and data management practices.
- Expertise in parallel programming (multi-threading, multi-processing, and distributed computing frameworks).
- Proficient in machine learning frameworks like PyTorch; familiarity with JAX, PyTorch Lightning, and GPU optimization.
- Hands-on experience with MLOps tools (e.g., Docker, Kubernetes) and cloud infrastructure (AWS, Azure, Google Cloud).
- Strong communication skills and ability to thrive in a fast-paced startup environment.
Benefits:
- Competitive salary with equity (BSPCE).
- Comprehensive health insurance
- Paid time off and vacation following French standards.
- Dynamic work environment with flexible, occasional remote options.
- Opportunity to work on cutting-edge AI technology that makes a real impact.
If you're passionate about using AI for sustainability and want to be part of a dynamic startup environment, we'd love to hear from you!