mlxrs 0.1.0

Safe Rust bindings for Apple's MLX array framework, with LM, VLM, audio, and embeddings support
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

//! Method-form comparison bridges.

use crate::{array::Array, error::Result};

impl Array {
  /// Element-wise equality. See [`crate::ops::comparison::equal`].
  pub fn equal(&self, rhs: &Array) -> Result<Array> {
    crate::ops::comparison::equal(self, rhs)
  }

  /// Element-wise inequality. See [`crate::ops::comparison::not_equal`].
  pub fn not_equal(&self, rhs: &Array) -> Result<Array> {
    crate::ops::comparison::not_equal(self, rhs)
  }

  /// Element-wise strict less-than. See [`crate::ops::comparison::less`].
  pub fn less(&self, rhs: &Array) -> Result<Array> {
    crate::ops::comparison::less(self, rhs)
  }

  /// Element-wise less-than-or-equal. See [`crate::ops::comparison::less_equal`].
  pub fn less_equal(&self, rhs: &Array) -> Result<Array> {
    crate::ops::comparison::less_equal(self, rhs)
  }

  /// Element-wise strict greater-than. See [`crate::ops::comparison::greater`].
  pub fn greater(&self, rhs: &Array) -> Result<Array> {
    crate::ops::comparison::greater(self, rhs)
  }

  /// Element-wise greater-than-or-equal. See [`crate::ops::comparison::greater_equal`].
  pub fn greater_equal(&self, rhs: &Array) -> Result<Array> {
    crate::ops::comparison::greater_equal(self, rhs)
  }

  /// Scalar-Bool "all-close" check. See [`crate::ops::comparison::allclose`].
  pub fn allclose(&self, rhs: &Array, rtol: f64, atol: f64, equal_nan: bool) -> Result<Array> {
    crate::ops::comparison::allclose(self, rhs, rtol, atol, equal_nan)
  }

  /// Element-wise close-to check. See [`crate::ops::comparison::isclose`].
  pub fn isclose(&self, rhs: &Array, rtol: f64, atol: f64, equal_nan: bool) -> Result<Array> {
    crate::ops::comparison::isclose(self, rhs, rtol, atol, equal_nan)
  }

  /// Element-wise finite check. See [`crate::ops::comparison::isfinite`].
  pub fn isfinite(&self) -> Result<Array> {
    crate::ops::comparison::isfinite(self)
  }

  /// Element-wise inf check. See [`crate::ops::comparison::isinf`].
  pub fn isinf(&self) -> Result<Array> {
    crate::ops::comparison::isinf(self)
  }

  /// Element-wise NaN check. See [`crate::ops::comparison::isnan`].
  pub fn isnan(&self) -> Result<Array> {
    crate::ops::comparison::isnan(self)
  }

  /// Element-wise negative-infinity check. See [`crate::ops::comparison::isneginf`].
  pub fn isneginf(&self) -> Result<Array> {
    crate::ops::comparison::isneginf(self)
  }

  /// Element-wise positive-infinity check. See [`crate::ops::comparison::isposinf`].
  pub fn isposinf(&self) -> Result<Array> {
    crate::ops::comparison::isposinf(self)
  }
}