Skip to main content

Module raw

Module raw

Re-exports§

pub use self::mlx_dtype_ as mlx_dtype;
pub use self::mlx_device_type_ as mlx_device_type;
pub use self::mlx_compile_mode_ as mlx_compile_mode;

Structs§

Constants§

Statics§

mlx_array_empty^⚠

Functions§

_mlx_array_is_available^⚠
_mlx_array_is_col_contiguous^⚠
_mlx_array_is_contiguous^⚠
_mlx_array_is_row_contiguous^⚠
_mlx_array_wait^⚠
_mlx_error^⚠
mlx_abs^⚠
mlx_add^⚠
mlx_addmm^⚠
mlx_all^⚠
mlx_all_axes^⚠
mlx_all_axis^⚠
mlx_allclose^⚠
mlx_any^⚠
mlx_any_axes^⚠
mlx_any_axis^⚠
mlx_arange^⚠
mlx_arccos^⚠
mlx_arccosh^⚠
mlx_arcsin^⚠
mlx_arcsinh^⚠
mlx_arctan^⚠
mlx_arctan2^⚠
mlx_arctanh^⚠
mlx_argmax^⚠
mlx_argmax_axis^⚠
mlx_argmin^⚠
mlx_argmin_axis^⚠
mlx_argpartition^⚠
mlx_argpartition_axis^⚠
mlx_argsort^⚠
mlx_argsort_axis^⚠
mlx_array_data_bool^⚠
mlx_array_data_complex64^⚠
mlx_array_data_float32^⚠
mlx_array_data_float64^⚠
mlx_array_data_int8^⚠
mlx_array_data_int16^⚠
mlx_array_data_int32^⚠
mlx_array_data_int64^⚠
mlx_array_data_uint8^⚠
mlx_array_data_uint16^⚠
mlx_array_data_uint32^⚠
mlx_array_data_uint64^⚠
mlx_array_dim^⚠
mlx_array_dtype^⚠
mlx_array_equal^⚠
mlx_array_eval^⚠
mlx_array_free^⚠
mlx_array_item_bool^⚠
mlx_array_item_complex64^⚠
mlx_array_item_float32^⚠
mlx_array_item_float64^⚠
mlx_array_item_int8^⚠
mlx_array_item_int16^⚠
mlx_array_item_int32^⚠
mlx_array_item_int64^⚠
mlx_array_item_uint8^⚠
mlx_array_item_uint16^⚠
mlx_array_item_uint32^⚠
mlx_array_item_uint64^⚠
mlx_array_itemsize^⚠
mlx_array_nbytes^⚠
mlx_array_ndim^⚠
mlx_array_new^⚠
mlx_array_new_bool^⚠
mlx_array_new_complex^⚠
mlx_array_new_data^⚠
mlx_array_new_data_managed^⚠
mlx_array_new_data_managed_payload^⚠
mlx_array_new_double^⚠
mlx_array_new_float^⚠
mlx_array_new_float32^⚠
mlx_array_new_float64^⚠
mlx_array_new_int^⚠
mlx_array_set^⚠
mlx_array_set_bool^⚠
mlx_array_set_complex^⚠
mlx_array_set_data^⚠
mlx_array_set_double^⚠
mlx_array_set_float^⚠
mlx_array_set_float32^⚠
mlx_array_set_float64^⚠
mlx_array_set_int^⚠
mlx_array_shape^⚠
mlx_array_size^⚠
mlx_array_strides^⚠
mlx_array_tostring^⚠
mlx_as_strided^⚠
mlx_astype^⚠
mlx_async_eval^⚠
mlx_atleast_1d^⚠
mlx_atleast_2d^⚠
mlx_atleast_3d^⚠
mlx_bartlett^⚠
mlx_bitwise_and^⚠
mlx_bitwise_invert^⚠
mlx_bitwise_or^⚠
mlx_bitwise_xor^⚠
mlx_blackman^⚠
mlx_block_masked_mm^⚠
mlx_broadcast_arrays^⚠
mlx_broadcast_to^⚠
mlx_ceil^⚠
mlx_checkpoint^⚠
mlx_clear_cache^⚠
mlx_clip^⚠
mlx_closure_apply^⚠
mlx_closure_custom_apply^⚠
mlx_closure_custom_free^⚠
mlx_closure_custom_jvp_apply^⚠
mlx_closure_custom_jvp_free^⚠
mlx_closure_custom_jvp_new^⚠
mlx_closure_custom_jvp_new_func^⚠
mlx_closure_custom_jvp_new_func_payload^⚠
mlx_closure_custom_jvp_set^⚠
mlx_closure_custom_new^⚠
mlx_closure_custom_new_func^⚠
mlx_closure_custom_new_func_payload^⚠
mlx_closure_custom_set^⚠
mlx_closure_custom_vmap_apply^⚠
mlx_closure_custom_vmap_free^⚠
mlx_closure_custom_vmap_new^⚠
mlx_closure_custom_vmap_new_func^⚠
mlx_closure_custom_vmap_new_func_payload^⚠
mlx_closure_custom_vmap_set^⚠
mlx_closure_free^⚠
mlx_closure_kwargs_apply^⚠
mlx_closure_kwargs_free^⚠
mlx_closure_kwargs_new^⚠
mlx_closure_kwargs_new_func^⚠
mlx_closure_kwargs_new_func_payload^⚠
mlx_closure_kwargs_set^⚠
mlx_closure_new^⚠
mlx_closure_new_func^⚠
mlx_closure_new_func_payload^⚠
mlx_closure_new_unary^⚠
mlx_closure_set^⚠
mlx_closure_value_and_grad_apply^⚠
mlx_closure_value_and_grad_free^⚠
mlx_closure_value_and_grad_new^⚠
mlx_closure_value_and_grad_new_func^⚠
mlx_closure_value_and_grad_new_func_payload^⚠
mlx_closure_value_and_grad_set^⚠
mlx_compile^⚠
mlx_concatenate^⚠
mlx_concatenate_axis^⚠
mlx_conjugate^⚠
mlx_contiguous^⚠
mlx_conv1d^⚠
mlx_conv2d^⚠
mlx_conv3d^⚠
mlx_conv_general^⚠
mlx_conv_transpose1d^⚠
mlx_conv_transpose2d^⚠
mlx_conv_transpose3d^⚠
mlx_copy^⚠
mlx_cos^⚠
mlx_cosh^⚠
mlx_cuda_is_available^⚠
mlx_cummax^⚠
mlx_cummin^⚠
mlx_cumprod^⚠
mlx_cumsum^⚠
mlx_custom_function^⚠
mlx_custom_vjp^⚠
mlx_default_cpu_stream_new^⚠
mlx_default_gpu_stream_new^⚠
mlx_degrees^⚠
mlx_depends^⚠
mlx_dequantize^⚠
mlx_detail_compile^⚠
mlx_detail_compile_clear_cache^⚠
mlx_detail_compile_erase^⚠
mlx_detail_vmap_replace^⚠
mlx_detail_vmap_trace^⚠
mlx_device_count^⚠
mlx_device_equal^⚠
mlx_device_free^⚠
mlx_device_get_index^⚠
mlx_device_get_type^⚠
mlx_device_info_free^⚠
mlx_device_info_get^⚠
mlx_device_info_get_keys^⚠
mlx_device_info_get_size^⚠
mlx_device_info_get_string^⚠
mlx_device_info_has_key^⚠
mlx_device_info_is_string^⚠
mlx_device_info_new^⚠
mlx_device_is_available^⚠
mlx_device_new^⚠
mlx_device_new_type^⚠
mlx_device_set^⚠
mlx_device_tostring^⚠
mlx_diag^⚠
mlx_diagonal^⚠
mlx_disable_compile^⚠
mlx_distributed_all_gather^⚠
mlx_distributed_all_max^⚠
mlx_distributed_all_min^⚠
mlx_distributed_all_sum^⚠
mlx_distributed_group_free^⚠
mlx_distributed_group_new^⚠
mlx_distributed_group_rank^⚠
mlx_distributed_group_size^⚠
mlx_distributed_group_split^⚠
mlx_distributed_init^⚠
mlx_distributed_is_available^⚠
mlx_distributed_recv^⚠
mlx_distributed_recv_like^⚠
mlx_distributed_send^⚠
mlx_distributed_sum_scatter^⚠
mlx_divide^⚠
mlx_divmod^⚠
mlx_dtype_size^⚠
mlx_einsum^⚠
mlx_enable_compile^⚠
mlx_equal^⚠
mlx_erf^⚠
mlx_erfinv^⚠
mlx_eval^⚠
mlx_exp^⚠
mlx_expand_dims^⚠
mlx_expand_dims_axes^⚠
mlx_expm1^⚠
mlx_export_function^⚠
mlx_export_function_kwargs^⚠
mlx_export_to_dot^⚠
mlx_eye^⚠
mlx_fast_cuda_kernel_apply^⚠
mlx_fast_cuda_kernel_config_add_output_arg^⚠
mlx_fast_cuda_kernel_config_add_template_arg_bool^⚠
mlx_fast_cuda_kernel_config_add_template_arg_dtype^⚠
mlx_fast_cuda_kernel_config_add_template_arg_int^⚠
mlx_fast_cuda_kernel_config_free^⚠
mlx_fast_cuda_kernel_config_new^⚠
mlx_fast_cuda_kernel_config_set_grid^⚠
mlx_fast_cuda_kernel_config_set_init_value^⚠
mlx_fast_cuda_kernel_config_set_thread_group^⚠
mlx_fast_cuda_kernel_config_set_verbose^⚠
mlx_fast_cuda_kernel_free^⚠
mlx_fast_cuda_kernel_new^⚠
mlx_fast_layer_norm^⚠
mlx_fast_metal_kernel_apply^⚠
mlx_fast_metal_kernel_config_add_output_arg^⚠
mlx_fast_metal_kernel_config_add_template_arg_bool^⚠
mlx_fast_metal_kernel_config_add_template_arg_dtype^⚠
mlx_fast_metal_kernel_config_add_template_arg_int^⚠
mlx_fast_metal_kernel_config_free^⚠
mlx_fast_metal_kernel_config_new^⚠
mlx_fast_metal_kernel_config_set_grid^⚠
mlx_fast_metal_kernel_config_set_init_value^⚠
mlx_fast_metal_kernel_config_set_thread_group^⚠
mlx_fast_metal_kernel_config_set_verbose^⚠
mlx_fast_metal_kernel_free^⚠
mlx_fast_metal_kernel_new^⚠
mlx_fast_rms_norm^⚠
mlx_fast_rope^⚠
mlx_fast_rope_dynamic^⚠
mlx_fast_scaled_dot_product_attention^⚠
mlx_fft_fft^⚠
mlx_fft_fft2^⚠
mlx_fft_fftn^⚠
mlx_fft_fftshift^⚠
mlx_fft_ifft^⚠
mlx_fft_ifft2^⚠
mlx_fft_ifftn^⚠
mlx_fft_ifftshift^⚠
mlx_fft_irfft^⚠
mlx_fft_irfft2^⚠
mlx_fft_irfftn^⚠
mlx_fft_rfft^⚠
mlx_fft_rfft2^⚠
mlx_fft_rfftn^⚠
mlx_flatten^⚠
mlx_floor^⚠
mlx_floor_divide^⚠
mlx_from_fp8^⚠
mlx_full^⚠
mlx_full_like^⚠
mlx_function_exporter_apply^⚠
mlx_function_exporter_apply_kwargs^⚠
mlx_function_exporter_free^⚠
mlx_function_exporter_new^⚠
mlx_gather^⚠
mlx_gather_mm^⚠
mlx_gather_qmm^⚠
mlx_gather_single^⚠
mlx_get_active_memory^⚠
mlx_get_cache_memory^⚠
mlx_get_default_device^⚠
mlx_get_default_stream^⚠
mlx_get_memory_limit^⚠
mlx_get_peak_memory^⚠
mlx_greater^⚠
mlx_greater_equal^⚠
mlx_hadamard_transform^⚠
mlx_hamming^⚠
mlx_hanning^⚠
mlx_identity^⚠
mlx_imag^⚠
mlx_imported_function_apply^⚠
mlx_imported_function_apply_kwargs^⚠
mlx_imported_function_free^⚠
mlx_imported_function_new^⚠
mlx_inner^⚠
mlx_io_gguf_free^⚠
mlx_io_gguf_get_array^⚠
mlx_io_gguf_get_keys^⚠
mlx_io_gguf_get_metadata_array^⚠
mlx_io_gguf_get_metadata_string^⚠
mlx_io_gguf_get_metadata_vector_string^⚠
mlx_io_gguf_has_metadata_array^⚠
mlx_io_gguf_has_metadata_string^⚠
mlx_io_gguf_has_metadata_vector_string^⚠
mlx_io_gguf_new^⚠
mlx_io_gguf_set_array^⚠
mlx_io_gguf_set_metadata_array^⚠
mlx_io_gguf_set_metadata_string^⚠
mlx_io_gguf_set_metadata_vector_string^⚠
mlx_io_reader_descriptor^⚠
mlx_io_reader_free^⚠
mlx_io_reader_new^⚠
mlx_io_reader_tostring^⚠
mlx_io_writer_descriptor^⚠
mlx_io_writer_free^⚠
mlx_io_writer_new^⚠
mlx_io_writer_tostring^⚠
mlx_isclose^⚠
mlx_isfinite^⚠
mlx_isinf^⚠
mlx_isnan^⚠
mlx_isneginf^⚠
mlx_isposinf^⚠
mlx_jvp^⚠
mlx_kron^⚠
mlx_left_shift^⚠
mlx_less^⚠
mlx_less_equal^⚠
mlx_linalg_cholesky^⚠
mlx_linalg_cholesky_inv^⚠
mlx_linalg_cross^⚠
mlx_linalg_eig^⚠
mlx_linalg_eigh^⚠
mlx_linalg_eigvals^⚠
mlx_linalg_eigvalsh^⚠
mlx_linalg_inv^⚠
mlx_linalg_lu^⚠
mlx_linalg_lu_factor^⚠
mlx_linalg_norm^⚠
mlx_linalg_norm_l2^⚠
mlx_linalg_norm_matrix^⚠
mlx_linalg_pinv^⚠
mlx_linalg_qr^⚠
mlx_linalg_solve^⚠
mlx_linalg_solve_triangular^⚠
mlx_linalg_svd^⚠
mlx_linalg_tri_inv^⚠
mlx_linspace^⚠
mlx_load^⚠
mlx_load_gguf^⚠
mlx_load_reader^⚠
mlx_load_safetensors^⚠
mlx_load_safetensors_reader^⚠
mlx_log^⚠
mlx_log2^⚠
mlx_log1p^⚠
mlx_log10^⚠
mlx_logaddexp^⚠
mlx_logcumsumexp^⚠
mlx_logical_and^⚠
mlx_logical_not^⚠
mlx_logical_or^⚠
mlx_logsumexp^⚠
mlx_logsumexp_axes^⚠
mlx_logsumexp_axis^⚠
mlx_map_string_to_array_free^⚠
mlx_map_string_to_array_get^⚠
mlx_map_string_to_array_insert^⚠
mlx_map_string_to_array_iterator_free^⚠
mlx_map_string_to_array_iterator_new^⚠
mlx_map_string_to_array_iterator_next^⚠
mlx_map_string_to_array_new^⚠
mlx_map_string_to_array_set^⚠
mlx_map_string_to_string_free^⚠
mlx_map_string_to_string_get^⚠
mlx_map_string_to_string_insert^⚠
mlx_map_string_to_string_iterator_free^⚠
mlx_map_string_to_string_iterator_new^⚠
mlx_map_string_to_string_iterator_next^⚠
mlx_map_string_to_string_new^⚠
mlx_map_string_to_string_set^⚠
mlx_masked_scatter^⚠
mlx_matmul^⚠
mlx_max^⚠
mlx_max_axes^⚠
mlx_max_axis^⚠
mlx_maximum^⚠
mlx_mean^⚠
mlx_mean_axes^⚠
mlx_mean_axis^⚠
mlx_median^⚠
mlx_meshgrid^⚠
mlx_metal_is_available^⚠
mlx_metal_start_capture^⚠
mlx_metal_stop_capture^⚠
mlx_min^⚠
mlx_min_axes^⚠
mlx_min_axis^⚠
mlx_minimum^⚠
mlx_moveaxis^⚠
mlx_multiply^⚠
mlx_nan_to_num^⚠
mlx_negative^⚠
mlx_node_namer_free^⚠
mlx_node_namer_get_name^⚠
mlx_node_namer_new^⚠
mlx_node_namer_set_name^⚠
mlx_not_equal^⚠
mlx_number_of_elements^⚠
mlx_ones^⚠
mlx_ones_like^⚠
mlx_outer^⚠
mlx_pad^⚠
mlx_pad_symmetric^⚠
mlx_partition^⚠
mlx_partition_axis^⚠
mlx_power^⚠
mlx_print_graph^⚠
mlx_prod^⚠
mlx_prod_axes^⚠
mlx_prod_axis^⚠
mlx_put_along_axis^⚠
mlx_qqmm^⚠
mlx_quantize^⚠
mlx_quantized_matmul^⚠
mlx_radians^⚠
mlx_random_bernoulli^⚠
mlx_random_bits^⚠
mlx_random_categorical^⚠
mlx_random_categorical_num_samples^⚠
mlx_random_categorical_shape^⚠
mlx_random_gumbel^⚠
mlx_random_key^⚠
mlx_random_laplace^⚠
mlx_random_multivariate_normal^⚠
mlx_random_normal^⚠
mlx_random_normal_broadcast^⚠
mlx_random_permutation^⚠
mlx_random_permutation_arange^⚠
mlx_random_randint^⚠
mlx_random_seed^⚠
mlx_random_split^⚠
mlx_random_split_num^⚠
mlx_random_truncated_normal^⚠
mlx_random_uniform^⚠
mlx_real^⚠
mlx_reciprocal^⚠
mlx_remainder^⚠
mlx_repeat^⚠
mlx_repeat_axis^⚠
mlx_reset_peak_memory^⚠
mlx_reshape^⚠
mlx_right_shift^⚠
mlx_roll^⚠
mlx_roll_axes^⚠
mlx_roll_axis^⚠
mlx_round^⚠
mlx_rsqrt^⚠
mlx_save^⚠
mlx_save_gguf^⚠
mlx_save_safetensors^⚠
mlx_save_safetensors_writer^⚠
mlx_save_writer^⚠
mlx_scatter^⚠
mlx_scatter_add^⚠
mlx_scatter_add_axis^⚠
mlx_scatter_add_single^⚠
mlx_scatter_max^⚠
mlx_scatter_max_single^⚠
mlx_scatter_min^⚠
mlx_scatter_min_single^⚠
mlx_scatter_prod^⚠
mlx_scatter_prod_single^⚠
mlx_scatter_single^⚠
mlx_segmented_mm^⚠
mlx_set_cache_limit^⚠
mlx_set_compile_mode^⚠
mlx_set_default_device^⚠
mlx_set_default_stream^⚠
mlx_set_error_handler^⚠
mlx_set_memory_limit^⚠
mlx_set_wired_limit^⚠
mlx_sigmoid^⚠
mlx_sign^⚠
mlx_sin^⚠
mlx_sinh^⚠
mlx_slice^⚠
mlx_slice_dynamic^⚠
mlx_slice_update^⚠
mlx_slice_update_dynamic^⚠
mlx_softmax^⚠
mlx_softmax_axes^⚠
mlx_softmax_axis^⚠
mlx_sort^⚠
mlx_sort_axis^⚠
mlx_split^⚠
mlx_split_sections^⚠
mlx_sqrt^⚠
mlx_square^⚠
mlx_squeeze^⚠
mlx_squeeze_axes^⚠
mlx_squeeze_axis^⚠
mlx_stack^⚠
mlx_stack_axis^⚠
mlx_std^⚠
mlx_std_axes^⚠
mlx_std_axis^⚠
mlx_stop_gradient^⚠
mlx_stream_equal^⚠
mlx_stream_free^⚠
mlx_stream_get_device^⚠
mlx_stream_get_index^⚠
mlx_stream_new^⚠
mlx_stream_new_device^⚠
mlx_stream_set^⚠
mlx_stream_tostring^⚠
mlx_string_data^⚠
mlx_string_free^⚠
mlx_string_new^⚠
mlx_string_new_data^⚠
mlx_string_set^⚠
mlx_subtract^⚠
mlx_sum^⚠
mlx_sum_axes^⚠
mlx_sum_axis^⚠
mlx_swapaxes^⚠
mlx_synchronize^⚠
mlx_take^⚠
mlx_take_along_axis^⚠
mlx_take_axis^⚠
mlx_tan^⚠
mlx_tanh^⚠
mlx_tensordot^⚠
mlx_tensordot_axis^⚠
mlx_tile^⚠
mlx_to_fp8^⚠
mlx_topk^⚠
mlx_topk_axis^⚠
mlx_trace^⚠
mlx_transpose^⚠
mlx_transpose_axes^⚠
mlx_tri^⚠
mlx_tril^⚠
mlx_triu^⚠
mlx_unflatten^⚠
mlx_value_and_grad^⚠
mlx_var^⚠
mlx_var_axes^⚠
mlx_var_axis^⚠
mlx_vector_array_append_data^⚠
mlx_vector_array_append_value^⚠
mlx_vector_array_free^⚠
mlx_vector_array_get^⚠
mlx_vector_array_new^⚠
mlx_vector_array_new_data^⚠
mlx_vector_array_new_value^⚠
mlx_vector_array_set^⚠
mlx_vector_array_set_data^⚠
mlx_vector_array_set_value^⚠
mlx_vector_array_size^⚠
mlx_vector_int_append_data^⚠
mlx_vector_int_append_value^⚠
mlx_vector_int_free^⚠
mlx_vector_int_get^⚠
mlx_vector_int_new^⚠
mlx_vector_int_new_data^⚠
mlx_vector_int_new_value^⚠
mlx_vector_int_set^⚠
mlx_vector_int_set_data^⚠
mlx_vector_int_set_value^⚠
mlx_vector_int_size^⚠
mlx_vector_string_append_data^⚠
mlx_vector_string_append_value^⚠
mlx_vector_string_free^⚠
mlx_vector_string_get^⚠
mlx_vector_string_new^⚠
mlx_vector_string_new_data^⚠
mlx_vector_string_new_value^⚠
mlx_vector_string_set^⚠
mlx_vector_string_set_data^⚠
mlx_vector_string_set_value^⚠
mlx_vector_string_size^⚠
mlx_vector_vector_array_append_data^⚠
mlx_vector_vector_array_append_value^⚠
mlx_vector_vector_array_free^⚠
mlx_vector_vector_array_get^⚠
mlx_vector_vector_array_new^⚠
mlx_vector_vector_array_new_data^⚠
mlx_vector_vector_array_new_value^⚠
mlx_vector_vector_array_set^⚠
mlx_vector_vector_array_set_data^⚠
mlx_vector_vector_array_set_value^⚠
mlx_vector_vector_array_size^⚠
mlx_version^⚠
mlx_view^⚠
mlx_vjp^⚠
mlx_where^⚠
mlx_zeros^⚠
mlx_zeros_like^⚠

Type Aliases§