U

Unconventional AI

Analog computing substrate for energy-efficient AI

System Modeling - Dynamic Systems Simulation

Software EngineeringConfirmed live in the last 24 hours
Full-timeNo salary listed

About the Job

Requirements

  • Knowledge of analog and mixed-signal circuit design, including transistor-level circuit design principles and modeling nonidealities such as noise, mismatch, and process variations.
  • Proficiency in PyTorch or JAX, including building custom autograd functions and integrating numerical solvers such as Neural ODEs to represent dynamic processes.
  • Strong theoretical and practical grasp of linear and nonlinear dynamics, state-space representations, and solving dx/dt = f(x, u, t) within a machine-learning context.
  • Understanding of modeling and mitigating noise in real-world systems, including experience with stochastic differential equations or Bayesian filtering.
  • Industry experience building high-fidelity circuit simulations that balance computational efficiency with physical accuracy.
  • Solid understanding of modern AI/ML architectures and training/inference workflows.
  • Strong experience implementing and debugging ML models in PyTorch or similar, with practical experience profiling, optimizing, and stabilizing non-trivial large-scale ML systems.

Responsibilities

  • Develop high-performance PyTorch or JAX components that model complex, time-varying circuit-based dynamic systems.
  • Build custom, high-speed, high-fidelity behavioral simulators to enable rapid iteration across physics-based computing systems.
  • Develop physics-based system models, GPU-accelerated ML system simulations, and cross-layer system integration.

Desired Qualifications

  • Familiarity with hardware-level concepts such as circuit dynamics, signal processing, or transfer functions.
  • Strong Python engineering skills, including modular design, testing, packaging, and continuous integration.
  • Experience with PyTorch internals, including autograd, custom modules, low-level operations, and familiarity with torch.compile or similar graph-capture/compile flows.
  • Experience with CUDA, Triton, or other GPU programming approaches, including writing custom kernels, understanding memory hierarchy, and basic performance tuning.
  • Comfort with some of JAX, NumPy, TensorFlow, Modal, high-performance computing patterns such as MPI and NCCL or distributed training, and SciPy.
  • Ability to reason across algorithm, software, runtime, and hardware layers.
  • Ability to connect model architecture choices to system performance implications, including memory bandwidth, communication patterns, latency, energy, and numerical issues.
  • Experience applying efficiency techniques such as quantization, sparsity, pruning, distillation, or kernel fusion.

About the Company

Unconventional AI

Analog computing substrate for energy-efficient AI

Unconventional AI designs analog AI hardware and silicon substrates that run neural networks directly in hardware. Its chips use analog circuits that mimic biological neuron dynamics to process AI workloads in the physics of silicon, reducing energy use compared with digital GPUs. It differentiates itself by pursuing an in-hardware, neuromorphic-style substrate rather than optimizing existing GPUs, supported by prominent venture funding. Its goal is to counter the AI energy bottleneck and enable scalable AI in data centers with biology-scale efficiency.

Company size
51-200
Company stage
Seed
Total funding
$475M
Headquarters
San Diego, CA
Founded
2025

Recently Posted Jobs

Explore more opportunities at Unconventional AI

See more jobs