Software Engineer Torrance, California, United States
Indexed description
Backed by premier venture capital and led by engineers and scientists who’ve landed rockets and pioneered novel AI forecast techniques, we move fast, own the hard problems end to end, and measure success in real world impact. If you want to build frontier tech with outsized consequences, welcome aboard.
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
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
- Experience with numerical weather prediction, remote-sensing data, or geospatial intelligence
- Background in physics-informed ML, simulation modeling, or data assimilation
- Contributions to open-source ML or scientific computing frameworks
- Small, senior team; high autonomy, high expectations.
- Collaborate closely with the founding team and SMEs to push the boundaries of observation-driven AI-native weather forecasting, and turn weather into an advantage, not a roadblock.
- LA HQ required.
- Equity upside commensurate with being one of the first core team members. Competitive salary, full benefits, and hardware budget.
Create a free Caio profile to unlock more results and save your role and location preferences.
Unlock free search