Machine Learning Engineer
Root Accessabout 2 hours ago
New York, NY, USAMid Level / Senior
Responsibilities
- Design and train deep learning models optimized for solving Maxwell’s equations and other physics equations.
- Develop high-performance asset pipelines for converting PCB files into tensor grids or graph embeddings.
- Implement Differentiable Physics Calibration pipelines to fine-tune material and manufacturing parameters using lab measurements.
- Collaborate on integrating Graph Neural Networks with spatial physics engines.
- Optimize training and inference pipelines for real-time execution on GPU clusters.
Requirements
- Master’s or Ph.D. in Computer Science, Mathematics, EE, Physics, or a related quantitative field.
- 4+ years of expert-level experience with PyTorch or JAX.
- Hands-on experience building and training Physics-Informed Neural Networks (PINNs) or Fourier Neural Operators (FNOs).
- Exceptional understanding of partial differential equations, vector calculus, and numerical optimization algorithms.
- Strong proficiency in manipulating spatial or geometric datasets using Python libraries.