Senior Software Engineer

Posted 14 Days Ago
Santa Clara, CA
175K-225K Annually
Senior level
Artificial Intelligence • Machine Learning
The Role
As a Senior Software Engineer at Celestial AI, you will develop production-grade firmware for AI solutions using Photonic Fabric technology. Your role involves gathering requirements, implementing communication software, and collaborating closely with hardware/software teams, as well as ensuring high-quality output through best practices in software development.
Summary Generated by Built In

About Celestial AI

As Generative AI continues to advance, the performance drivers for data center infrastructure are shifting from systems-on-chip (SOCs) to systems of chips. In the era of Accelerated Computing, data center bottlenecks are no longer limited to compute performance, but rather the system’s interconnect bandwidth, memory bandwidth, and memory capacity. Celestial AI’s Photonic Fabric™ is the next-generation interconnect technology that delivers a tenfold increase in performance and energy efficiency compared to competing solutions.

The Photonic Fabric™ is available to our customers in multiple technology offerings, including optical interface chiplets, optical interposers, and Optical Multi-chip Interconnect Bridges (OMIB). This allows customers to easily incorporate high bandwidth, low power, and low latency optical interfaces into their AI accelerators and GPUs. The technology is fully compatible with both protocol and physical layers, including standard 2.5D packaging processes. This seamless integration enables XPUs to utilize optical interconnects for both compute-to-compute and compute-to-memory fabrics, achieving bandwidths in the tens of terabits per second with nanosecond latencies.

This innovation empowers hyperscalers to enhance the efficiency and cost-effectiveness of AI processing by optimizing the XPUs required for training and inference, while significantly reducing the TCO2 impact. To bolster customer collaborations, Celestial AI is developing a Photonic Fabric ecosystem consisting of tier-1 partnerships that include custom silicon/ASIC design, system integrators, HBM memory, assembly, and packaging suppliers.

ABOUT THE ROLE

As Senior Software Engineer, you will be a key player in the full lifecycle development of an AI platform system, including requirements, design, code, and test. In this role, you will be part of a team developing next generation AI/ML solutions using the Celestial AI Photonic Fabric Architecture. You will be responsible for developing production-grade firmware for Celestial AI’s photonic fabric product lineup and defining and developing firmware APIs for resource management and optimization. You will collaborate closely with hardware and software teams to implement the communication link management software for different Photonic Fabric use cases.  

ESSENTIAL DUTIES AND RESPONSIBILITIES

  • Participate in gathering technical requirements, contributing to software design specifications
  • Provide support to both internal and external development teams
  • Actively engage in code reviews, discussions to share knowledge, and improve code quality
  • Perform best practices in software development and documentation to ensure designs meet requirements and to deliver high-quality software on tight schedules.

QUALIFICATIONS

  • Bachelor’s degree with 8+ years of experience, or Master’s degree with 6+ years of experience in Computer Science, Electrical Engineering, Information Technology or a related technical field
  • Strong C/C++, Python programming skills with software design skills, including debugging, performance analysis, and test design
  • Experience with:
    • Baseboard management controller (BMC) software
    • Open-source software - use of GitHub, bug tracking, branching and merging code, OSS licensing issues handling patches, etc.
  • Familiarity with:
    • Networking protocols including PCIe, Infiniband, and Ethernet
    • Memory controllers such as HBM, DDR, and other on-chip memory interfaces
    • SoC interfaces including AXI and other chip-to-chip interfaces
  • Strong analytical, debugging, and testing methodology skills with ability to efficiently debug, root cause, and resolve issues spanning multiple subsystems
  • Ability to work in a fast-paced, agile team environment 

For California location:

As an early startup experiencing explosive growth, we offer an extremely attractive total compensation package, inclusive of competitive base salary and a generous grant of our valuable early-stage equity. The target base salary for this role is approximately $175,000.00 - $225,000.00. The base salary offered may be slightly higher or lower than the target base salary, based on the final scope as determined by the depth of the experience and skills demonstrated by candidate in the interviews.

We offer great benefits (health, vision, dental and life insurance), collaborative and continuous learning work environment, where you will get a chance to work with smart and dedicated people engaged in developing the next generation architecture for high performance computing.

Celestial AI Inc. is proud to be an equal opportunity workplace and is an affirmative action employer.


#LI-Onsite

Top Skills

C
C++
Python
The Company
HQ: Santa Clara, CA
60 Employees
On-site Workplace
Year Founded: 2020

What We Do

Celestial AI is a Machine Learning (ML) accelerator company that has developed a proprietary technology platform which enables the next generation of high-performance computing solutions. Celestial AI’s mission is to transform data parallel computing with a proprietary Photonic Fabric™ technology platform which uses light for data movement both within chip and between chips.

Advancements in data communications have driven robust silicon photonics technology and volume manufacturing ecosystems that are ripe for commercial implementation of ML and high-performance computing (HPC) solutions which leverage integrated silicon photonics for data movement.

Celestial AI’s system delivers differentiated single node performance that scales efficiently, providing significant performance gains for multi-node and multi model applications. The scalability of Celestial AI’s accelerator architecture enables an efficient and performant mapping of data and compute over a broad range of ML model types without the need for complex software optimizations. Celestial AI’s competitive advantage will further grow over time as ML models continue to increase in complexity and size.

Celestial AI has assembled a highly experienced team of industry leaders who have a track record of building multiple successful technology businesses. The company’s Orion AI accelerator products serve an addressable market that is projected by Omida to exceed $70 billion in 2025

Similar Jobs

Easy Apply
3 Locations
1100 Employees

Finch Logo Finch

Senior Software Engineer, Fullstack

Fintech • HR Tech • Software
Hybrid
2 Locations
68 Employees
180K-200K Annually

VSCO Logo VSCO

Senior Software Engineer

Computer Vision • Digital Media • Mobile • Productivity • Social Media • Software
Easy Apply
Remote
2 Locations
110 Employees

Check Logo Check

Senior Software Engineer

Fintech • HR Tech • Payments • Infrastructure as a Service (IaaS) • Data Privacy
Remote
San Francisco, CA, USA
100 Employees

Similar Companies Hiring

Halter Thumbnail
Software • Machine Learning • Internet of Things • Hardware • Greentech • Business Intelligence • Agriculture
Auckland City, NZ
150 Employees
InCommodities Thumbnail
Renewable Energy • Machine Learning • Information Technology • Energy • Automation • Analytics
Austin, TX
234 Employees
RunPod Thumbnail
Software • Infrastructure as a Service (IaaS) • Cloud • Artificial Intelligence
Charlotte, North Carolina
53 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account