Team Lead, AI Software Engineering - Remote

WHAT'S THE DAY TO DAY LIKE?

• Ideate, project manage, and execute against DLWP research tasks.
• Manage the “buy or build” question as it relates to adopting and adapting open source DLWP models.
• Train, verify and deploy DLWP models on enormous amounts of data using SOTA on-prem and cloud accelerators.
• Champion technical designs, recruit feedback, code and configure, project manage, and document ML systems, data, and pre/post processing pipelines.
• Mentor and lead meteorologists, climatologists, data scientists, and software engineers. 
• Help the engineering organization transition to “AI first”.
• Bring a production software mindset to research and development tasks. 
• Conduct and solicit code reviews.
• Create a culture of inclusivity, psychological safety, honesty, and vulnerability.
• Coordinate roadmaps with key product and research stakeholders.
• Physical Effort: No physical effort required.
• Business Travel: Infrequent travel to Oakville, Canada, may be required.

WE ARE HOPING YOU HAVE:

• 7+ years of professional experience.
• At least 1 year of formal leadership experience, or clear demonstration of leadership interest and abilities, including strong interest in teaching and mentorship.


Meteorological and/or AI background:
• Coursework or published research in weather/climate forecasting, artificial intelligence, or both.


Hands-on AI experience:
• Experience with all parts of the AI lifecycle, including training and deploying data-hungry deep learning models in a commercial context.
• Scientific Python: Xarray, Dask, Numpy, Scipy, Pandas, Scikit-learn.
• ML Python: JAX, Haiku, PyTorch, TensorFlow, Caffe, Keras.
• Low-level memory management frameworks: OpenMPI, CUDA, XLA.
• Debugging skills: training and learning/loss dynamics, time/memory/utilization profiling, multi-process debugging, and log management.  
• Distributed training experience: muti-node, multi-accelerator environments.
• Knowledge of modern DL architectures and how they apply to weather/climate prediction problems.


Generalist software engineering and cloud infrastructure skills:
• Coding: Object Oriented Programming, Design Patterns, multithreading/processing, git, code profiling, debugging. 
• Cloud: Airflow/Dagster, Kubernetes, Helm, Terraform, Kubeflow, container registries, CI/CD, log management, data version control. 
• Linux: process management, telemetry, IO, bash scripting, environment configuration.
• Traditional NWP/HPC experience a plus: WRF, MPAS, OpenMP, Slurm, Parallel Cluster, OpenACC, Infiniband, Fortran.


Experience with weather and climate data.
• Grib, NetCDF, Arrow IPC, Parquet, Zarr and supporting libraries (Eccodes, Xarray, etc).
• Spatial and temporal interpolation techniques.
• Knowledge of forecast and climate models, parameters, and verification techniques.


Exceptional leadership skills:
• Communication, presentation, project management, including technical writing and design skills. 
• Influential: leverages personal and team experience to guide decision making diplomatically and effectively.
• Optimistic, open minded, inclusive.

YOU ARE MADE FOR THIS IF YOU...

• Think “at scale”: default to documentation, automation, source control, force multiplication, etc.
• Have hands-on DLWP experience with one/some of: GraphCast, MetNet, ClimaX, FourCastNet, SEEDS, GenCast, Pangu Weather, FuXi, WeatherBench, CorrDiff, …




DON’T MEET ALL THE REQUIREMENTS? DON’T WORRY…
Not all applicants will have skills that match a job description exactly. Pelmorex values and welcomes diverse experiences. While having “desired” qualifications make for a strong candidate, we encourage applicants with alternative experiences to also apply. If your career is just starting or hasn't followed a traditional path, don't let that stop you from considering Pelmorex. We are always looking for people who will be a culture add (yup, that's right, we don’t subscribe to culture fit, we want you to add to what we think is an amazing culture and work environment

INTERESTED? IT GETS BETTER….

• Course Reimbursement Program 📚 We want you to keep learning, so we can too
• Open and transparent communication, including All Hands Meetings with our CEO
• Pelmorex Learning Academy 🏫 includes offerings like French, Leadership (for people leaders and non-leaders alike),  yoga, mindfulness
• Your mental health is important to us! 🧠 We partner with Inkblot for virtual counseling sessions
• Frequent employee pulse surveys 📝 we value your feedback so we can continue to make Pelmorex a great place to work!
• Free online doctor visits 👩‍⚕️ with Maple Online Healthcare
• While we encourage 1:1 conversations, we recognize that not everyone is comfortable with speaking up 🗣 We have an anonymous reporting platform (Speakfully) to ensure everyone’s voice is heard