عملیات و سخت‌افزار پشتیبانی‌شده

ما در حال حاضر از هسته‌های پردازنده گرافیکی زیر پشتیبانی می‌کنیم:

• tokamax.dot_product_attention ( توجه سریع ).
• tokamax.gated_linear_unit ( واحدهای خطی دروازه‌دار (SwiGLU و غیره)).
• tokamax.layer_norm ( نرمال‌سازی لایه‌ها و نرمال‌سازی جذر میانگین مربعات ).

و موارد زیر برای هر دو GPU و TPU:

• tokamax.ragged_dot ( ترکیبی از متخصصان ).