mlx-native-0.3.2
- mlx-native 0.3.2
- Docs.rs crate page
- MIT
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- robertelee78
- Dependencies
- bytemuck ^1 normal
  half ^2 normal
  memmap2 ^0.9 normal
  metal ^0.33 normal
  objc ^0.2 normal
  safetensors ^0.7 normal
  serde ^1 normal
  serde_json ^1 normal
  thiserror ^2 normal
  criterion ^0.5 dev
  static_assertions ^1 dev
- Versions
- 95.56% of the crate is documented
Platform
- aarch64-apple-darwin
- x86_64-apple-darwin
Feature flags

docs.rs

Rust

Skip to main content

register

mlx_native0.3.2

In mlx_native::ops::argmax

mlx_native::ops::argmax

Function register

pub fn register(registry: &mut KernelRegistry)

Expand description

Register argmax shader source with the given kernel registry.