New opportunity

Machine Learning Engineer

Octave-X · Chicago, United States

About this role

Responsible for end-to-end lifecycle of AI models for Tenzin, including research prototyping, designing and optimizing distributed training and inference pipelines, building evaluation suites and observability, and contributing to formal verification layers for enterprise deployments in regulated domains.

Skills for this role

PythonPyTorchJAXDistributed trainingMulti-GPU trainingMulti-node trainingModel parallelismInference optimizationTransformer architecturesAttention mechanismsLarge language model evaluationMLOpsExperiment trackingModel versioningCI/CD for MLMonitoringObservabilityData drift detectionFormal verificationHomotopy Type TheoryLeanCoqAgdaGPU performance profilingKernel optimizationCloud-native toolingTraining pipelinesInference pipelinesCommunication