Here is a list of all functions with links to the classes they belong to:

- a -

• Abs() : mlx::core::Abs
• Add() : mlx::core::Add
• AddMM() : mlx::core::AddMM
• advance() : pocketfft::detail::multi_iter< N >, pocketfft::detail::rev_iter, pocketfft::detail::simple_iter
• AffineQuantize() : mlx::core::fast::AffineQuantize
• aligned_allocator() : pocketfft::detail::threading::aligned_allocator< T >
• AllGather() : mlx::core::distributed::AllGather
• allocate() : pocketfft::detail::threading::aligned_allocator< T >
• Allocator() : mlx::core::allocator::Allocator
• AllReduce() : mlx::core::distributed::AllReduce
• apply() : mlx::steel::TransformAdd< OutT, InT >, mlx::steel::TransformAxpby< OutT, InT >, mlx::steel::TransformNone< OutT, InT >, ScaleOp< OutT, InT >
• apply_epilogue() : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
• apply_epilogue_safe() : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
• apply_inplace_op() : mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >
• Arange() : mlx::core::Arange
• ArcCos() : mlx::core::ArcCos
• ArcCosh() : mlx::core::ArcCosh
• ArcSin() : mlx::core::ArcSin
• ArcSinh() : mlx::core::ArcSinh
• ArcTan() : mlx::core::ArcTan
• ArcTan2() : mlx::core::ArcTan2
• ArcTanh() : mlx::core::ArcTanh
• ArgPartition() : mlx::core::ArgPartition
• ArgReduce() : mlx::core::ArgReduce
• ArgSort() : mlx::core::ArgSort
• argument_encoder() : mlx::core::metal::Device
• arr() : pocketfft::detail::arr< T >
• arr_info() : pocketfft::detail::arr_info
• array() : mlx::core::array
• ArrayIterator() : mlx::core::array::ArrayIterator
• AsStrided() : mlx::core::AsStrided
• AsType() : mlx::core::AsType
• atomic_update() : None
• attach_event() : mlx::core::array