Back to list

Vacancy description

Principal Machine Learning Engineer - Rust & AI Infrastructure in भारत

IT

भारत

About the Role :I'm looking for a Principal Machine Learning Engineer (Rust) to build out the infrastructure layer for an unconventional, exascale AI pre-training https://jobeax.com/link/7xHW0abXorsagaHJ we are building :We are completely moving away from traditional, centralized error backpropagation and its massive All-Reduce network bottlenecks. Instead, we are implementing a highly distributed, gradient-free alternative inspired by Hinton's Forward-Forward https://jobeax.com/link/1oXo7eplLp7sVkGT vertically slice model parameters across a distributed network and stream token batches horizontally. Our stack is entirely built in Rust, leveraging Burn for deep learning math, Zenoh for low-latency asynchronous data transport, and Tokio for https://jobeax.com/link/OC1Qgbjfj124B7TI protect our IP and our clients' data, we deploy this stack using Confidential Containers (CoCo), which wraps our runtime in a hardware-encrypted micro-VM blanket (AMD SEV-SNP, Arm CCA, and Nvidia Confidential Computing) straight to the silicon https://jobeax.com/link/r0u3onxSynEmnhUD combining this stateless hub-and-edge matrix math with CoCo's zero-copy memory pinning, we can bypass standard OS serialization taxes and keep processing units at maximum compute saturation. We can pre-train a Llama 3.1 405B scale model on a fixed 460-node cluster in weeks, not https://jobeax.com/link/LpkP65zy7KUaLGGV Current Status :- Stage: Pre-revenue, validation, and core architecture design phase.- Hardware: Planned architecture footprint (Inspur aggregators, Supermicro hubs with H100s, and Arm/Nvidia edge configurations).- Funding/Path: We are weighing two routes: dropping this straight into a infrastructure-seed-round pitch for VC funding (the TTM numbers look incredible) OR leveraging our speed advantages to secure a signed, paying enterprise design partner while we finish the https://jobeax.com/link/qfGNHuKrLD0TAKbs we are looking for :You are a Principal-level Rust engineer who eats low-level systems programming for breakfast. You aren't an "AI wrapper" developer - you understand how hardware actually works. Specifically, we need deep competence in:- Advanced Rust: Safe concurrency, async programming (Tokio), and custom memory management (zero-allocation arenas, memory pinning).- Linux Internals & Networking: High-throughput network primitives, zero-copy architecture, gRPC/byte-streaming, and ideally RDMA.- Infrastructural Security: Familiarity with virtualization, container runtimes (Kata), and hardware TEEs (AMD SEV, Intel TDX, Arm CCA) is a massive https://jobeax.com/link/heFokhD3hFStDVXU Experience : 1 - 3 years (ref:hirist.tech)

Similar vacancies

More like this role

Kaliper Technologies Pvt Ltd

Machine Learning Engineer

  • Remote
  • Contract
  • AI / ML

$60 USD - $120 USD / hour

... evaluations, improving model performance, and building reliable training and inference workflows. Role Overview Develop, implement, review, and validate complex machine learning engineering solutions. The role requires strong Python skills, practical machine learning experience, and hands-on use of coding agents in development ...

Lucknow, भारत

  • Full-time
  • AI / ML
... Own the full model lifecycle : data collection, feature engineering, training, validation, deployment, and monitoring.- Fine-tune LLMs and smaller models for domain-specific tasks, structured extraction from credit reports, and negotiation dialogue quality.- Build and maintain the evaluation framework that catches model quality ...

Bangalore, भारत

  • Hybrid
  • On-site
  • Full-time
  • Contract
  • IT
... mentoring engineers to uphold technical excellence. What you’ll do (Role Expectations) - Design and build a low-latency, high-throughput data forwarding plane using Rust and async/await models for efficient service-oriented infrastructure - Develop distributed, scalable systems with a heavy focus on concurrency, fault tolerance, ...

Bangalore, भारत

  • AI / ML
Amgen seeks a Senior Machine Learning Engineer to design and scale end-to-end ML and generative AI platforms. The role involves building enterprise-grade infrastructure, MLOps services, and governance controls that enable data scientists and practitioners to develop, deploy, and monitor models securely at scale. Responsibilities ...

Hyderabad, भारत

  • Hybrid
  • Contract
  • AI / ML
Job Description :Role & Responsibilities :ML & AI System Design and Development :- Design and implement machine learning solutions using a mix of classical ML models and LLM-based approaches.- Select appropriate techniques for each problem, balancing accuracy, performance, cost, and operational complexity.- Build and train ...

Noida, भारत

  • AI / ML
Amgen seeks a Senior Machine Learning Engineer to develop scalable ML platforms and workflows supporting drug discovery and development. This role involves productionizing AI/ML solutions, implementing MLOps best practices, and collaborating with scientists and engineers across the organization in Hyderabad, India. Responsibilities ...

Hyderabad, भारत

  • Full-time
  • AI / ML
... training at scale : multi-GPU or multi-node training (DDP, FSDP, DeepSpeed, or equivalent).- Strong fundamentals in linear algebra, probability, optimisation, and machine learning.- Strong software engineering https://jobeax.com/link/dUQ1ixxrbyA0NLAm Stack : C++, Rust, Golang, TypeScript, Jetson, ONNX, TensorRTAbout PhyAgents ...

Bangalore, भारत

  • Hybrid
  • AI / ML
... information, please visit . Overall Step into the role of Machine Learning Engineer IV at Brightstar in Hyderabad, Telangana, India. Guide the progress of our AI and machine learning solutions, championing innovation and excellence. To excel as a Machine Learning Engineer IV, you need: - Extensive Python programming experience ...

Hyderabad, भारत

  • Remote
  • Full-time
... collection of vast datasets relating to the changes that software developers make in source code when working in an enterprise software development environment. As a Machine Learning Data Engineer you will take problems and ideas from our clients and customer support teams, analyse what would be involved, whether the solution will ...

Bangalore, भारत

  • Hybrid
  • On-site
  • Full-time
  • Contract
  • AI / ML
... architects to drive platform adoption. Qualifications - Experience: 9+ years of progressive, highly relevant experience in software engineering and machine learning development. - Production Excellence: A proven, demonstrable track record of successfully architecting, building, and productionizing complex Machine Learning ...

Bangalore, भारत

  • Contract
  • AI / ML
... optimization- Pricing and margin optimization- Operations planning and routingWhat you'll do :- Build and deploy ML models end to end: data preparation, feature engineering, training, evaluation, deployment, and monitoring.- Own a problem area, from deciding what's worth building to the business metrics it moves.- Write production-quality ...

Bangalore, भारत