hub / github.com/dromara/Omega-AI

github.com/dromara/Omega-AI @main

Chat with this repo

repository ↗ · DeepWiki ↗ · + Follow

13,903 symbols 45,757 edges 882 files 911 documented · 7%

What it actually does AI analysis from the code graph — generated when you open this

loading…

Extension points exported contracts — how you extend this code

Tokenizer (Interface)

(no doc) [7 implementers]

src/main/java/com/omega/example/transformer/tokenizer/bertTokenizer/Tokenizer.java

Core symbols most depended-on inside this repo

get

called by 3703

src/main/java/com/omega/engine/tensor/Tensor.java

loadData

called by 3010

src/main/java/com/omega/example/common/ModeLoaderlUtils.java

getGpuData

called by 2670

src/main/java/com/omega/engine/tensor/Tensor.java

Tensor_OP

called by 2042

src/main/java/com/omega/engine/nn/layer/Layer.java

createGPUTensor

called by 1498

src/main/java/com/omega/engine/tensor/Tensor.java

size

called by 981

src/main/java/com/omega/common/utils/MatrixUtils.java

add

called by 964

src/main/java/com/omega/engine/nn/layer/gpu/BasicBlockKernel.java

setInput

called by 825

src/main/java/com/omega/engine/nn/layer/Layer.java

Shape

Method 13,024

Class 857

Enum 21

Interface 1

Languages

Java100%

Modules by API surface

src/main/java/com/omega/engine/tensor/Tensor.java102 symbols

src/main/java/com/omega/engine/optimizer/MBSGDOptimizer.java99 symbols

src/main/java/com/omega/common/utils/MatrixOperation.java88 symbols

src/main/java/com/omega/engine/ad/op/gpu/OPKernel.java86 symbols

src/main/java/com/omega/engine/ad/op/TensorOP.java75 symbols

src/main/java/com/omega/engine/nn/layer/normalization/gpu/BNKernel.java53 symbols

src/main/java/com/omega/engine/nn/network/Network.java50 symbols

src/main/java/com/omega/engine/gpu/BNKernel.java49 symbols

src/main/java/com/omega/common/data/LayerConfig.java45 symbols

src/main/java/com/omega/engine/nn/layer/normalization/gpu/BNKernel2.java41 symbols

src/main/java/com/omega/example/dit/test/DiTTest.java40 symbols

src/main/java/com/omega/engine/ad/Graph.java40 symbols

For agents

$ claude mcp add Omega-AI \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact