Senior Software Engineer - Machine Learning
About this role
- We cannot sponsor or transfer any visas, of any kind, at this time*
- The salary offered for this position will be based on a candidate’s experience and skill demonstrated during interviews and other evaluations
- Framework parity and behavioral correctness. Help close feature and semantic gaps against scikit-learn and Spark ML across the model catalog so our models behave the way users coming from those frameworks expect
- New model types and capabilities. Expand the model catalog across classification, regression, time series, and automated model selection, including loss functions and objectives we don't support today.
- Performance at scale. Make training and inference fast on datasets that don't fit anywhere else. Approximate nearest-neighbor acceleration, distributed optimizer tuning, and algorithmic work on the hot paths.
- Numerical and linear-algebra foundations. Help expand the SQL-level linear algebra surface (SVD, eigenvalues, matrix inverse and solve, sparse matrices) and spectral transforms, the substrate under PCS, regression, and optimization.
- Architecture. Our models are compiled into the query plan and execute as a native part of it, rather than running in a separate ML runtime. You'll work inside that architecture and help improve it.
- Collaboration and craft. Write clear design docs, tests, and documentation; investigate issues where behavior diverges from user expectations; and partner with Product, architects, and customer-facing teams to identify gaps before customers hit them.
- 5+ years building production software systems, including solid experience in C++ (or comparable systems-level work in Java/Scala with a willingness to work primarily in C++).
- Hands-on experience implementing or integrating machine learning models in production. You have written the training loop, not just called into a library.
- Working knowledge of numerical methods: gradient-based optimization, loss functions and their gradients, numerical stability, feature scaling, convergence behavior.
- Familiarity with scikit-learn, Spark ML, XGBoost, or comparable frameworks, and awareness of where their defaults and semantics matter.
- Strong instincts around correctness, edge cases, and behavioral consistency – and the discipline to encode them in tests and documentation.
- Ability to work across teams and codebases and turn ambiguous requirements into concrete solutions.
- Experience comparing or validating model behavior across multiple ML frameworks.
- Experience with large-scale data systems, analytical databases, query planners, or distributed execution engines.
- Exposure to optimization (LP/QP/SOCP), spectral methods (FFT/DCT/DWT), causal inference, or probabilistic modeling for the in-database research surface we are building next.
- Experience with automatic differentiation or symbolic gradient generation.
- Familiarity with SQL internals like AST manipulation, expression rewriting, or planner integration.
- Customers see fewer surprises. Our models behave the way someone coming from scikit-learn or Spark ML expects, and where they differ, the difference is intentional and documented.
- Roadmap items such as ARIMA, quantile regression, AutoML, option parity, land with validated correctness against reference implementations.
- Feature gaps are identified from benchmarks and product analysis early, not discovered under customer pressure.
- You deliver across both parity work and broader ML initiatives, balancing short-term needs with long-term quality.
Company at a glance
For the organizations that can't get AI at enterprise scale wrong, there's Ocient.
The OcientAIQ™ ecosystem delivers trusted agentic AI solutions at petabyte scale, built for the industries where the data is largest, the stakes are high, and production-grade results are non-negotiable. Where fragmented data stacks slow agents down and drive up costs, OcientAIQ™ brings the infrastructure and expertise to make it work.
Ocient is a global, remote-first, carbon-neutral company, headquartered in Chicago, and backed by leading investors including Greycroft, OCA Ventures, In-Q-Tel and Buoyant Ventures.
Our remote-first company is made up of collaborative, curious, and driven experts who are passionate about delivering solutions that drive real-world impact. If that sounds like you, check out our open jobs or reach out to [email protected].
Tired of cold applications?
Sign up with Clera and we'll reach out the moment a role actually fits you — no more spraying applications into the void.
Know someone who'd be great for this?