diff --git a/mlx/transforms.h b/mlx/transforms.h index ff47638bf..caf648163 100644 --- a/mlx/transforms.h +++ b/mlx/transforms.h @@ -80,7 +80,7 @@ ValueAndGradFn value_and_grad( /** * Returns a function which computes the value and gradient of the input - * function with repsect to a single input array. + * function with respect to a single input array. **/ ValueAndGradFn inline value_and_grad( const std::function(const std::vector&)>& fun, @@ -132,7 +132,7 @@ std::function(const std::vector&)> inline grad( /** * Returns a function which computes the gradient of the input function with - * repsect to a single input array. + * respect to a single input array. * * The function being differentiated takes a vector of arrays and returns an * array. The optional `argnum` index specifies which the argument to compute