Make AI run on
every device

We're a research group building the future of on-device AI. Private, fast, and free from cloud dependency.

Open Roles

AI Talent Product

Define and shape edge AI products that bring powerful, private AI directly to users' devices.

Beijing, China / Hefei Anhui, China

We Hope You

  • Have an engineering background
  • Deeply understand the edge AI landscape — cloud vs on-device trade-offs, privacy-first design, and user-centric AI
  • Love reading — stay current with AI research and industry trends
  • Have read at least 30 AI papers (100+ if born after 1998)
  • Can translate cutting-edge research into practical product roadmaps
  • Passionate about making AI accessible, private, and free for everyone
Apply Now

Edge AI Research Engineer

Research and develop model compression techniques to make state-of-the-art AI models run efficiently on consumer devices.

Beijing, China / Hefei Anhui, China

Tech Stack

PythonPyTorchMLXONNX

Focus

  • Model compression: quantization (FP16→INT8→INT4), structured/unstructured pruning, and knowledge distillation.
  • Designing efficient model architectures optimized for edge deployment (1B–4B parameters).
  • Benchmarking inference performance across consumer hardware (Apple Silicon, Snapdragon, etc.).
  • Reproducing and improving state-of-the-art methods from deep learning literature.
  • Publishing research and contributing to open-source projects.

Ideal Experience

  • Deep understanding of Transformer architecture and large language models.
  • Hands-on experience with model compression techniques (quantization, pruning, distillation).
  • Familiarity with efficient architectures: MobileLLM, Phi, Gemma, and similar.
  • Performance profiling and optimization on resource-constrained devices.
  • Track record of open-source contributions or published research.
Apply Now

Apple MLX Engineer

Optimize the MLX framework and build high-performance ML infrastructure on Apple Silicon.

Beijing, China / Hefei Anhui, China

Tech Stack

PythonMLXMetalSwiftC++

Focus

  • Optimizing MLX framework internals for maximum performance on M-series chips.
  • Memory optimization: mmap acceleration, zero-copy model loading, unified memory utilization.
  • Integrating MLX with Apple frameworks (Core ML, Metal Performance Shaders).
  • Profiling and debugging ML workloads on Apple Silicon.
  • Building inference pipelines for on-device LLM, VLM, and TTS models.

Ideal Experience

  • Hands-on development with the MLX framework.
  • Low-level performance optimization on Apple hardware (M1–M5).
  • GPU programming with Metal or similar compute APIs.
  • Memory management and system-level optimization.
  • Contributing to open-source ML frameworks.
Apply Now

Swift Product Engineer

Build native iOS/macOS consumer applications powered by on-device AI for the EchoStream AI product line.

Beijing, China / Hefei Anhui, China

Tech Stack

SwiftSwiftUICore MLMLXXcode

Focus

  • Building native consumer apps: on-device translation, multimodal finance, food analysis, and more.
  • Integrating on-device ML models (LLM, VLM, TTS) into production iOS/macOS applications.
  • Optimizing app performance, memory footprint, and battery efficiency with AI workloads.
  • Designing delightful user experiences that leverage AI capabilities transparently.
  • Shipping and iterating products in the App Store.

Ideal Experience

  • Building and shipping polished iOS/macOS applications with Swift and SwiftUI.
  • Integrating Core ML or MLX models into native apps.
  • Performance optimization on mobile/desktop: memory, CPU, GPU usage.
  • Strong product sense — experience building user-facing consumer products.
  • Familiarity with on-device ML inference workflows.
Apply Now

Full Stack Engineer

Build web platforms, research tooling, and developer experiences for AtomGradient's open-source ecosystem.

Beijing, China / Hefei Anhui, China

Tech Stack

ReactTypeScriptRustPythonNext.js

Focus

  • Building internal research tools, benchmarking dashboards, and model evaluation platforms.
  • Developing and maintaining AtomGradient's web presence and documentation sites.
  • Creating developer-facing tools for the open-source community.
  • End-to-end application development from backend services to frontend interfaces.

Ideal Experience

  • Building professional-grade web applications with React and TypeScript.
  • Backend service development in Rust or Python.
  • Experience with developer tools, documentation systems, or open-source project infrastructure.
  • Deploying and monitoring production web applications.
  • Interest in AI/ML and developer experience.
Apply Now

Don't see a fit?

We're always interested in hearing from people passionate about on-device AI and open-source research.

Get in touch