greenhouse
Software Engineer
Mantari
Long Beach, USA
Full-time
Senior · 5+ years experience
Hybrid
Discovered 1 weeks ago
CUDAJAXPyTorchTensorFlowKubernetesSlurm
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
CUDAJAXPyTorch
Key Skills for This Role
CUDAJAXPyTorchTensorFlowKubernetesSlurm
Full Job Posting
Description
Own the guts of our modeling engine, everything between raw sensor data and a lightning-fast forecast API:
Build and optimize CUDA kernels, JAX/PyTorch ops, and distributed training loops for physics-aware transformer models
Stand up reproducible, containerized pipelines (Ray, Kubernetes, Slurm, or similar) that scale from a single GPU to multi-region GPU clusters in the cloud
Automate data ingest and feature engineering from our growing sensor network (satellite, buoy, radar, etc.)
Translate novel research prototypes into production-grade code
You’ll work closely with Mantari’s Head of Weather and AI modeling teams to operationalize next-generation forecast systems, bringing meteorological science and modern ML together at planetary scale.
Requirements
- Degree in Atmospheric Science, Computer Science, or related engineering / physical sciences field
- 5+ years writing high-performance Python/C++/Rust for GPU workloads
- Real-world experience designing and building cloud-native applications and infrastructure (GCP, AWS, or Azure)
- Deep fluency with at least one modern deep learning framework (JAX, PyTorch, TensorFlow, etc.)
- Proven track record shipping distributed training or inference systems at scale (100+ GPUs)
- Comfortable profiling kernels, diagnosing memory bottlenecks, and reading compiler logs when needed
- Strong grasp of modern DevOps: code reviews, CI/CD pipelines, observability, and containerization
Bonus Qualifications
- Experience with numerical weather prediction, remote-sensing data, or geospatial intelligence
- Background in physics-informed ML, simulation modeling, or data assimilation
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer