About this role
Design, implement, and maintain large-scale model training systems and developer tooling to scale reinforcement learning and machine learning pipelines for Eka Robotics, collaborating with researchers and contributing to core JAX model and training code.
Skills for this role
JAXPyTorchTensorFlowDistributed trainingKubernetesSLURMGCPAWSCI/CD for ML workflowsLoggingTelemetryExperiment trackingCheckpointingJob orchestrationSchedulingResource managementData pipelinesDistributed systemsSoftware engineeringReinforcement learningRoboticsDeveloper experienceScalable systemsMulti-host setups