MCPcopy Create free account
hub / github.com/AtomaAI/atoma-infer

github.com/AtomaAI/atoma-infer @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
786 symbols 2,092 edges 58 files ⚖ Apache-2.0 224 documented · 28% updated 10d ago★ 13052 open issues

Browse by type

Functions 613 Types & classes 173
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

atoma-infer

Atoma Logo

atoma-infer is a Rust and CUDA project for large-language-model inference.

Learn more about Atoma: Visit atoma.ai for information about Atoma's secure AI infrastructure platform.

Status

The current implementation is under revival and is not production-ready. The repository does not yet have a verified build, test, or serving baseline. OpenAI API compatibility, model support, and single-node or distributed GPU topologies are not verified capabilities of this checkout. Do not rely on it for production workloads.

Rung 0 will restore a trustworthy build and test baseline before feature or performance claims are reintroduced. See the canonical rung-0 specification for its bounded scope.

Launch-gate parity target

Launch-gate parity is a future measured target, not a description of the current implementation. The gate requires DeepSeek-class goodput within 10% of the better of vLLM or SGLang, plus at least two outright headline benchmark wins. Measurements cover 2×8×H100 with FP8 and 8×B200 with NVFP4, in aggregated and prefill/decode-disaggregated topologies.

The canonical revival decision map records the decisions and measurement destination. The rung-0 specification defines the initial recovery work. These GitHub issues are the public sources of truth for the revival plan.

Contributor setup

  1. Fork the repository.
  2. Clone your fork: git clone https://github.com/YOUR-USERNAME/atoma-infer.git.
  3. Enter the checkout: cd atoma-infer.
  4. Install Rust using rustup. The repository's toolchain file selects the required Rust version.
  5. Initialize dependencies: git submodule update --init --recursive.

Verified build and test commands will be documented after rung 0 establishes the green baseline. Until then, failures from the inherited checkout are tracked by the rung-0 specification.

Contributing

Keep each pull request focused on one purpose, such as one bug fix, feature, or performance improvement. Unrelated changes belong in separate pull requests so each change can be reviewed independently.

A narrow exception applies to canonical roadmap integration pull requests. Such a pull request may integrate multiple planned changes only when it identifies their canonical tickets and preserves reviewable commit ranges for each ticket. This exception does not apply to unrelated cleanup or opportunistic changes.

Pull request descriptions should state the problem, the chosen approach, and the verification performed. Bug fixes and features should include tests at a public behavior seam. Performance changes should identify one bottleneck, describe the benchmark and hardware, and report speed and memory results.

License

Licensed under the Apache License 2.0.

Extension points exported contracts — how you extend this code

browse all types & interfaces →

Core symbols most depended-on inside this repo

browse all functions →

Shape

Method 360
Function 253
Class 125
Enum 40
Interface 8

Languages

Rust90%
C++10%

Modules by API surface

server/src/api/chat_completions.rs91 symbols
backends/vllm/src/sequence.rs88 symbols
backends/vllm/src/scheduler.rs79 symbols
backends/vllm/src/block_manager.rs33 symbols
csrc/src/lib.rs28 symbols
backends/vllm/src/config.rs28 symbols
backends/vllm/src/block.rs27 symbols
models/src/llama.rs26 symbols
csrc/kernels/utils.h26 symbols
csrc/tests/cache_manager_tests.rs23 symbols
models/src/flash_attention.rs21 symbols
backends/vllm/src/worker.rs19 symbols

For agents

$ claude mcp add atoma-infer \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact

Ask about this repo answers extend the page