Meshy is seeking a Machine Learning Engineer to enhance their 3D generative AI platform by implementing, integrating, and optimizing machine learning features, thereby improving the user experience in 3D creation.
Requirements
- Proficiency in Python and familiarity with ML frameworks (PyTorch, TensorFlow).
- Familiarity with software engineering best practices (unit testing, CI/CD, etc.).
- Hands-on experience (academic or industry) implementing ML models in PyTorch or TensorFlow and integrating them into applications.
- Familiarity with architecture and algorithms of modern generative models (transformers, diffusion, auto-regression).
- Experience with computer graphics or 3D modeling.
- Familiarity with cloud serving infrastructure and containerization (Docker, Kubernetes).
Responsibilities
- Develop ML-based product features that extend the capabilities of 3D foundation models; explore various solutions such as prompt engineering, lightweight fine-tuned models, or third-party API integrations.
- Collaborate with Research and Engineering teams to productionize internal machine learning models.
- Build unit and integration tests for ML components and ensure CI/CD pipelines run reliably
- Maintain clear documentation of model designs, API schemas, and deployment processes; participate in code reviews and design discussions.
- Implement machine learning solutions, such as prompt engineering, lightweight fine-tuned models, post-processing, that extend 3D foundation model inputs and outputs for 3D foundation models.
Other
- MS in Computer Science or related field or BS plus ~3 years of relevant experience.
