List of all items

Structs

• GraphOutputFile
• MlxArray
• MlxArrayVector
• MlxBindingsContext
• MlxClosure
• MlxClosureCustom
• MlxClosureCustomJvp
• MlxClosureCustomVmap
• MlxClosureKwargs
• MlxClosureValueAndGrad
• MlxCompileCache
• MlxCompiledDualHeadNorm
• MlxCompiledElementwiseGraphs
• MlxCompiledGatedDeltaRateActivation
• MlxCompiledMultiOutputGraph
• MlxCompiledSwiGlu
• MlxCudaKernel
• MlxCudaKernelConfig
• MlxDeviceHandle
• MlxDeviceInfo
• MlxDistributedGroup
• MlxFunctionExporter
• MlxImportedFunction
• MlxIoGguf
• MlxIoReader
• MlxIoWriter
• MlxMapStringToArray
• MlxMapStringToArrayIterator
• MlxMapStringToString
• MlxMapStringToStringIterator
• MlxMetalKernel
• MlxMetalKernelConfig
• MlxMetalKernelOutput
• MlxNodeNamer
• MlxStream
• MlxStreamThreadLocal
• MlxString
• MlxVectorInt
• MlxVectorStream
• MlxVectorString
• MlxVectorVectorArray
• error::MlxCError
• raw::__BindgenComplex
• raw::__BindgenFloat16
• raw::freestanding_file
• raw::mlx_array_
• raw::mlx_closure_
• raw::mlx_closure_custom_
• raw::mlx_closure_custom_jvp_
• raw::mlx_closure_custom_vmap_
• raw::mlx_closure_kwargs_
• raw::mlx_closure_value_and_grad_
• raw::mlx_compile_cache_
• raw::mlx_device_
• raw::mlx_device_info_
• raw::mlx_distributed_group_
• raw::mlx_fast_cuda_kernel_
• raw::mlx_fast_cuda_kernel_config_
• raw::mlx_fast_metal_kernel_
• raw::mlx_fast_metal_kernel_config_
• raw::mlx_function_exporter_
• raw::mlx_imported_function_
• raw::mlx_io_gguf_
• raw::mlx_io_reader_
• raw::mlx_io_vtable_
• raw::mlx_io_writer_
• raw::mlx_map_string_to_array_
• raw::mlx_map_string_to_array_iterator_
• raw::mlx_map_string_to_string_
• raw::mlx_map_string_to_string_iterator_
• raw::mlx_node_namer_
• raw::mlx_optional_dtype_
• raw::mlx_optional_float_
• raw::mlx_optional_int_
• raw::mlx_stream_
• raw::mlx_stream_thread_local_
• raw::mlx_string_
• raw::mlx_vector_array_
• raw::mlx_vector_int_
• raw::mlx_vector_stream_
• raw::mlx_vector_string_
• raw::mlx_vector_vector_array_

Enums

• MlxDtype
• MlxMetalKernelTemplateArgument

Functions

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

Type Aliases

• MlxGraphBuilder
• raw::FILE
• raw::bfloat16_t
• raw::float16_t
• raw::mlx_array
• raw::mlx_closure
• raw::mlx_closure_custom
• raw::mlx_closure_custom_jvp
• raw::mlx_closure_custom_vmap
• raw::mlx_closure_kwargs
• raw::mlx_closure_value_and_grad
• raw::mlx_compile_cache
• raw::mlx_compile_mode_
• raw::mlx_complex64_t
• raw::mlx_device
• raw::mlx_device_info
• raw::mlx_device_type_
• raw::mlx_distributed_group
• raw::mlx_dtype_
• raw::mlx_error_handler_func
• raw::mlx_fast_cuda_kernel
• raw::mlx_fast_cuda_kernel_config
• raw::mlx_fast_metal_kernel
• raw::mlx_fast_metal_kernel_config
• raw::mlx_fft_norm_
• raw::mlx_function_exporter
• raw::mlx_imported_function
• raw::mlx_io_gguf
• raw::mlx_io_reader
• raw::mlx_io_vtable
• raw::mlx_io_writer
• raw::mlx_map_string_to_array
• raw::mlx_map_string_to_array_iterator
• raw::mlx_map_string_to_string
• raw::mlx_map_string_to_string_iterator
• raw::mlx_node_namer
• raw::mlx_optional_dtype
• raw::mlx_optional_float
• raw::mlx_optional_int
• raw::mlx_stream
• raw::mlx_stream_thread_local
• raw::mlx_string
• raw::mlx_vector_array
• raw::mlx_vector_int
• raw::mlx_vector_stream
• raw::mlx_vector_string
• raw::mlx_vector_vector_array

Constants

• raw::BRIDGED_FUNCTIONS
• raw::BRIDGED_TYPES
• raw::mlx_compile_mode__MLX_COMPILE_MODE_DISABLED
• raw::mlx_compile_mode__MLX_COMPILE_MODE_ENABLED
• raw::mlx_compile_mode__MLX_COMPILE_MODE_NO_FUSE
• raw::mlx_compile_mode__MLX_COMPILE_MODE_NO_SIMPLIFY
• raw::mlx_device_type__MLX_CPU
• raw::mlx_device_type__MLX_GPU
• raw::mlx_dtype__MLX_BFLOAT16
• raw::mlx_dtype__MLX_BOOL
• raw::mlx_dtype__MLX_COMPLEX64
• raw::mlx_dtype__MLX_FLOAT16
• raw::mlx_dtype__MLX_FLOAT32
• raw::mlx_dtype__MLX_FLOAT64
• raw::mlx_dtype__MLX_INT16
• raw::mlx_dtype__MLX_INT32
• raw::mlx_dtype__MLX_INT64
• raw::mlx_dtype__MLX_INT8
• raw::mlx_dtype__MLX_UINT16
• raw::mlx_dtype__MLX_UINT32
• raw::mlx_dtype__MLX_UINT64
• raw::mlx_dtype__MLX_UINT8
• raw::mlx_fft_norm__MLX_FFT_NORM_BACKWARD
• raw::mlx_fft_norm__MLX_FFT_NORM_FORWARD
• raw::mlx_fft_norm__MLX_FFT_NORM_ORTHO