I am referencing some of the library functions under the max/nn directory on the modular github. For example I am trying to import: from nn.mha import _naive_attention_with_transpose, mha_gpu_naive from nn.mha_mask import CausalMask, MaterializedMask as one of the benchmark scripts does. I was ab…

Thanks for the prompt reply. I get this error when trying to follow the steps outlined in that post. When running ./bazelw build //max/kernels/src/nn, I get this error: //mojo/stdlib/stdlib:stdlib building mojo package failed: (Exit 1): mojo failed: error executing MojoPackage command (from target …

Importing functions from max.nn

MAX

Ehsan (Ehsan M. Kermani) July 11, 2025, 5:40am 2

For those Mojo modules, are you following this guide to build and import? I don’t think pixi can work with that as is.

Topic		Replies	Views
Importing MAX kernels from source General	2	172	July 11, 2025
MAX Nightly 25.4.0.dev2025061222 Released Nightly	0	43	June 13, 2025
How do I `import linalg`? GPU Programming	5	200	May 13, 2025
Mojo now available in Conda as a standalone package Mojo mojo-compiler	2	316	September 22, 2025
`pip install modular` now available! MAX	0	357	May 6, 2025

Importing functions from max.nn

Related topics