AI Engineer

• At Jua you will go on a mission to unriddle the world's atmosphere by using the latest deep learning and satellite technology to build tools for monitoring and forecasting. Renewable energy, precision agriculture, autonomous transport and parametric insurance will deliver on their potential only because of the work you do.
• Having raised millions in venture capital (at a sensible valuation), our first objective will be to provide businesses and countries with several orders of magnitude better and faster weather forecasts.
• You will be surrounded by people who have built several successful companies before and worked on exciting technologies used by hundreds of millions of people and thousands of businesses.
• You will contribute to our culture of an extremely high level of ambition, direct and friendly communication, rapid iteration and a healthy dose of humbleness. The team is completely remote with hubs currently in Zurich and Berlin.
• We offer exciting challenges, creative freedom, an extremely smart and cheerful team, fair compensation and generous shares in the company (and some other perks you would normally expect).

We are bridge builders and bring together the best in machine learning, physics, aerospace, data processing and UX design. Together, we will build one of the key components to enable a world worth living in at a time of changing climate and increasing extreme events.

What We Are Looking For

As a Senior AI Engineer, you will develop state-of-the-art weather forecasts based on machine learning and open & proprietary data for a sustainable world in times of climate change and increasing extreme weather. As part of our fast moving model team, you will be working on developing and improving our deep learning-based algorithms and have great creative freedom and influence. Based on the latest publications, you will test and implement state-of-the-art algorithms and approaches and, together with the team, build robust and sustainable machine learning pipelines for live operation.

Responsibilities And Tasks

• Developing and improving deep learning-based weather forecasts with state-of-the-art technologies
• Training these models on our GPU cluster Abacus
• Developing new algorithms from the idea through the prototype to live operation
• Testing, validating, visualizing, as well as documenting new approaches and their performance

Need-to-have

• You have an M.Sc., and preferably a PhD. in Computer Science or Machine Learning
• You have at least 5 years of professional experience in Machine Learning (ignore this point if you can proof in another way to have exceptional skills in that regard)
• You are proficient in Python
• You are familiar with the common machine learning frameworks (PyTorch or FLAX)
• You are very strong in linear algebra and have a clear understanding of the detailed processes going on in deep neural networks
• You are familiar with SOTA transformer architectures like GPT, Bart or Swin Transformer
• You have implemented at least one SOTA paper successfully from scratch by reading & understanding the paper
• You are familiar with common techniques like gradient clipping, layer parallelism or mixed precision to train very deep networks
• You are familiar with distributed training using both PyTorch build-ins as well as libraries like Huggingface Accelerate or Microsoft Deepspeed
• You have an intrinsic interest in improving our life on earth with technology
• You have experience in handling and training with large amounts of data (>1TB)
• You like to move fast, break things, take risks, think out of the box, iterate & learn fast
• You have experience working in the cloud

Nice-to-have

• You are familiar with experiment tracking tools (like WandB or ML flow)
• You have experience with GIS based data such as satellite, radar or radiosondes
• You have experience running Machine learning workloads in production environments

Only EMEA candidates will be considered at this stage.