Help us build
AI that enriches humanity

We are a team of AI researchers and engineers on a mission to build AI products that can help humanity enrich their lives. We are driven by ambitious goals, fast execution, and a strong sense of urgency. Join us if you want to shape the next generation of AI models and products.

Open Roles

We are always looking to hire the best people in the field. Explore the details below and apply directly if you're a great fit.

AI Talent Product

Beijing, China

We Hope You

  • Have an engineering background
  • Love reading
  • Have read at least 30 AI papers
  • If you were born after 1998, you need to read at least 100 AI papers

If so, please send your resume to alex@atomgradient.com, we will contact you asap.

Apply Now

Apple MLX Developer

Beijing, China

Tech Stack

  • Python
  • MLX
  • Metal
  • Swift (optionally)

Focus

  • Optimizing the MLX framework for high-performance machine learning on Apple Silicon.
  • Developing and testing new features for efficient model training and inference.
  • Integrating MLX with other Apple frameworks such as Core ML and Metal Performance Shaders.
  • Profiling and debugging ML workloads on M-series chips.

Ideal Experience

  • Hands-on experience developing with the MLX framework.
  • Performance optimization techniques for Apple hardware.
  • Building and deploying machine learning models on macOS or iOS.
  • Familiarity with GPU programming using Metal.
  • Experience contributing to open-source ML projects.
Apply Now

AI Engineer & Researcher

Beijing, China

Tech Stack

  • Rust
  • Python
  • JAX and XLA
  • CUDA (C++ and Triton)

Focus

  • Training trillion parameter neural networks at scale, as well as a variety of smaller specialized models.
  • Mastering all data modalities: text, audio, images, video, etc
  • Profiling, debugging, and optimizing multi-host GPU utilization.
  • Rapidly implementing the latest state of the art methods from deep learning literature.
  • Inventing new algorithms and methods that bring us closer to our goal: Developing AI systems that can accurately understand the universe and create new knowledge

Ideal Experience

  • Building distributed training systems for AI models in HPC clusters.
  • Configuring and troubleshooting hardware and operating-systems for maximum performance.
  • Working with JAX/XLA for multi-host training.
  • Writing custom CUDA kernels in either C++ or via Triton.
  • Digging into third-party source code for debugging and customization.
Apply Now

Data Engineering & Web Crawling

Beijing, China

Tech Stack

  • Rust
  • Python
  • Spark

Focus

  • Building petabyte-scale, high-throughput data processing systems.
  • Managing workloads across large cloud compute clusters.
  • Pre-processing datasets for AI training.

Ideal Experience

  • Building bespoke data processing libraries from scratch.
  • Designing and implementing distributed systems in Rust.
  • Keeping up with state-of-the-art techniques for preparing AI training data.
  • Organizing and meticulously bookkeeping data across multiple clouds, of multiple modalities, and from many sources.
Apply Now

Full Stack Web Engineer

Beijing, China

Tech Stack

  • React
  • Rust
  • gRPC
  • JS

Focus

  • Leveraging AT’s models and algorithms to create exceptionally useful and delightful tools and experiences.
  • Building end-to-end applications for both internal and external use.
  • Tinkering with new languages, libraries, tooling, and databases.

Ideal Experience

  • Building highly-performant and highly-available backend services using compiled languages.
  • Designing and coding polished, professional-grade applications in React.
  • Working with gRPC, REST APIs, and databases in Rust.
  • Upholding best practices for continuously deploying and monitoring production systems.
  • Securing applications and data using appropriate authentication and encryption measures.
Apply Now

Infrastructure Engineer

Beijing, China

Tech Stack

  • Kubernetes
  • Pulumi
  • Rust

Focus

  • Operating GPU supercomputing clusters for both AI training and serving production models.
  • Working with both on-premise clusters and cloud providers.
  • Help with security best practices for internal researchers and live external traffic.

Ideal Experience

  • Writing scalable and highly available containerized applications in Rust.
  • Managing compute fleets with Pulumi, Terraform, Ansible, or other stateful automation libraries.
  • Monitoring Kubernetes clusters with Prometheus, and Grafana.
  • Monitoring and root-causing faulty hardware, especially GPUs and RDMA networking.
  • Securing clusters via RBAC with OIDC or LDAP integrations, configuring cloud networking, and performing penetration testing.
Apply Now

Ready to Join?

Don't see a role that fits? Reach out to us directly. We're always interested in hearing from talented individuals.

Contact Us