3D Machine Learning Engineer

Job description

3D Machine Learning Engineer

About the job

Start date: 01.12.2024 or later - Six months contract

We are looking for a talented, driven, and overachieving individual to join our dedicated start-up team. In this role, you will have the exciting opportunity to contribute to the advancement of our cutting-edge geometry generation solutions. Your primary responsibility will be to set up 3D machine learning models and train them, making crucial decisions about data collection and labeling. Additionally, you will play a vital role in designing novel machine learning algorithms to tackle real-world problems in the high volume manufacturing industry.

In this role, you will …

1. Set up from the ground up 3D machine learning models, including design and training, while making autonomous decisions about data collection and labeling.
2. Design novel machine learning algorithms to solve real-world problems.
3. Build data pre- and post-processing pipelines for the machine learning models.
4. Optimize test-time ML pipelines for speed and memory usage.
5. Build processes to validate performance and robustness of the test-time ML pipelines.

In this role, you’ll need …

1. A background in Machine Learning / Computer Graphics.
2. 4+ years of relevant work experience.
3. Extensive experience in Julia and Rust.
4. Experience using geometric processing libraries such as openvdb, trimesh, opengl, cgal.
5. The ability to work on cutting-edge research projects while having a product mindset.
6. Knowledge about the most common neural network architectures such as AE, VAE, MLP, CNN, transformer.
7. Experience in bringing ML models to production (MLOps).
8. Some knowledge about 3D / geometric machine learning architectures.
9. Some knowledge about generative AI.

Nice to have …

1. Some experience with C/C++.
2. Some experience with ML frameworks, such as PyTorch, Flux.jl, etc.
3. Some experience with parallel computing / code optimization.
4. Some experience developing mathematical optimization algorithms.
5. Some experience in cloud computing / deployment including AWS, Docker, Apptainer.
6. Some knowledge of Italian.

Job Type: Temporary
Contract length: 6 months

Education:

• Master's (Preferred)

Experience:

• Software development: 3 years (Required)

Language:

• English (Required)