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

- s -

• sanity_check() : pocketfft::detail::util
• scalar_t : mlx::core::simd::Simd< T, N >, mlx::core::simd::Simd< float16_t, N >, mlx::core::simd::Simd< T, 1 >
• ScalarVector() : mlx::core::ScalarVector< Op >
• scale : mlx::steel::AttnParams, ScaleOp< OutT, InT >, TransformScale< T >
• ScaledDotProductAttention() : mlx::core::fast::ScaledDotProductAttention
• scales : QuantizedBlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, group_size, bits >
• Scan() : mlx::core::Scan
• Scatter() : mlx::core::Scatter
• ScatterAxis() : mlx::core::ScatterAxis
• scheduled : mlx::core::array
• Scheduler() : mlx::core::scheduler::Scheduler
• seed() : mlx::core::random::KeySequence
• seek() : mlx::core::ContiguousIterator, mlx::core::io::FileWriter, mlx::core::io::ParallelFileReader, mlx::core::io::Reader, mlx::core::io::Writer
• Select() : mlx::core::Select
• Send() : mlx::core::distributed::Send
• send() : mlx::core::distributed::detail::GroupImpl
• Set() : pocketfft::detail::cmplx< T >
• set_buffer() : mlx::core::CommandEncoder, mlx::core::metal::CommandEncoder
• set_bytes() : mlx::core::CommandEncoder, mlx::core::metal::CommandEncoder
• set_cache_limit() : mlx::core::metal::MetalAllocator
• set_compute_pipeline_state() : mlx::core::CommandEncoder, mlx::core::metal::CommandEncoder
• set_data() : mlx::core::array
• set_default_stream() : mlx::core::scheduler::Scheduler
• set_input_array() : mlx::core::CommandEncoder, mlx::core::metal::CommandEncoder
• set_memory_limit() : mlx::core::metal::MetalAllocator
• set_name() : mlx::core::NodeNamer
• set_output_array() : mlx::core::CommandEncoder, mlx::core::metal::CommandEncoder
• set_residency_set() : mlx::core::metal::Device
• set_siblings() : mlx::core::array
• set_status() : mlx::core::array
• set_tracer() : mlx::core::array
• set_value() : mlx::core::Event
• set_vector_bytes() : mlx::core::CommandEncoder, mlx::core::metal::CommandEncoder
• set_wired_limit() : mlx::core::metal::MetalAllocator
• shape() : mlx::core::array, mlx::core::fast::CustomKernelShapeInfo, mlx::core::ReductionPlan, mlx::steel::Layout2D< Shape, Layout >, pocketfft::detail::arr_info
• Shape2D() : mlx::steel::Shape2D< RInt, CInt >
• shapes : Indices< IdxT, NIDX >
• shp : pocketfft::detail::arr_info
• shutdown() : pocketfft::detail::threading::thread_pool
• siblings() : mlx::core::array
• Sigmoid() : mlx::core::Sigmoid
• Sign() : mlx::core::Sign
• signal() : mlx::core::Event
• signaling_NaN() : metal::_numeric_limits_impl< bfloat16_t >
• Simd() : mlx::core::simd::Simd< T, N >, mlx::core::simd::Simd< float16_t, N >, mlx::core::simd::Simd< T, 1 >
• simd_exclusive_scan() : CumMax< U >, CumMin< U >, CumProd< bool >
• simd_scan() : CumMax< U >, CumMin< U >, CumProd< bool >
• simple_iter() : pocketfft::detail::simple_iter
• Sin() : mlx::core::Sin
• sincos_2pibyn() : pocketfft::detail::sincos_2pibyn< T >
• Sinh() : mlx::core::Sinh
• size() : mlx::core::allocator::Allocator, mlx::core::allocator::CommonAllocator, mlx::core::array, mlx::core::distributed::detail::GroupImpl, mlx::core::distributed::Group, mlx::core::Dtype, mlx::core::metal::MetalAllocator, mlx::core::simd::Simd< T, N >, mlx::core::simd::Simd< float16_t, N >, mlx::core::simd::Simd< T, 1 >, pocketfft::detail::arr< T >, pocketfft::detail::arr_info
• Slice() : mlx::core::Slice
• SliceUpdate() : mlx::core::SliceUpdate
• sm : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
• sn : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
• Softmax() : mlx::core::Softmax
• Sort() : mlx::core::Sort
• sort() : BlockMergeSort< ValT, IdxT, ARG_SORT, BLOCK_THREADS, N_PER_THREAD, CompareOp >, ThreadSort< ValT, IdxT, ARG_SORT, N_PER_THREAD, CompareOp >
• special_mul() : pocketfft::detail::cmplx< T >
• Split() : mlx::core::Split
• split() : mlx::core::distributed::detail::GroupImpl, mlx::core::distributed::Group
• split_k_partition_size : mlx::steel::GEMMSpiltKParams
• split_k_partition_stride : mlx::steel::GEMMSpiltKParams
• split_k_partitions : mlx::steel::GEMMSpiltKParams
• Sqrt() : mlx::core::Sqrt
• Square() : mlx::core::Square
• Squeeze() : mlx::core::Squeeze
• src : mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >, mlx::steel::BlockLoaderT< T, BROWS, BCOLS, kDstStrRow, kDstStrCol, reduction_dim, tgp_size, n_reads, TCOLS, TROWS >, mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, QuantizedBlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, group_size, bits >
• src_ld : mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >, mlx::steel::BlockLoaderT< T, BROWS, BCOLS, kDstStrRow, kDstStrCol, reduction_dim, tgp_size, n_reads, TCOLS, TROWS >, mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, QuantizedBlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, group_size, bits >
• start_concurrent() : mlx::core::CommandEncoder, mlx::core::metal::CommandEncoder
• start_row : mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >
• state() : mlx::core::AddMM, mlx::core::Arange, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsStrided, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::BlockMaskedMM, mlx::core::Broadcast, mlx::core::BroadcastAxes, mlx::core::Cholesky, mlx::core::Concatenate, mlx::core::Convolution, mlx::core::DynamicSlice, mlx::core::DynamicSliceUpdate, mlx::core::Eigh, mlx::core::Equal, mlx::core::ExpandDims, mlx::core::fast::AffineQuantize, mlx::core::fast::LayerNormVJP, mlx::core::fast::RMSNormVJP, mlx::core::FFT, mlx::core::Flatten, mlx::core::Gather, mlx::core::GatherAxis, mlx::core::GatherQMM, mlx::core::Hadamard, mlx::core::Inverse, mlx::core::Log, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::QuantizedMatmul, mlx::core::RandomBits, mlx::core::Reduce, mlx::core::Reshape, mlx::core::Scan, mlx::core::Scatter, mlx::core::ScatterAxis, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Sqrt, mlx::core::Squeeze, mlx::core::Transpose, mlx::core::Unflatten, mlx::core::View
• Status : mlx::core::array
• status() : mlx::core::array
• step() : mlx::core::ContiguousIterator
• stop : mlx::core::scheduler::StreamThread
• StopGradient() : mlx::core::StopGradient
• store() : mlx::steel::BaseMMAFrag< T, 8, 8 >, mlx::steel::MMATile< T, kTileRows_, kTileCols_, MMAFrag_ >
• store_result() : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
• store_result_safe() : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
• store_safe() : mlx::steel::BaseMMAFrag< T, 8, 8 >, mlx::steel::MMATile< T, kTileRows_, kTileCols_, MMAFrag_ >
• str : MLXConvParams< NDIM >, pocketfft::detail::arr_info
• Stream() : mlx::core::Stream
• stream() : mlx::core::Event, mlx::core::Primitive, mlx::core::scheduler::StreamThread
• StreamContext() : mlx::core::StreamContext
• StreamThread() : mlx::core::scheduler::StreamThread
• stride() : pocketfft::detail::arr_info
• stride_in() : pocketfft::detail::multi_iter< N >
• stride_out() : pocketfft::detail::multi_iter< N >
• strided_device_idx : ReadWriter< in_T, out_T, step, four_step_real >
• strided_shared_idx : ReadWriter< in_T, out_T, step, four_step_real >
• strides : Indices< IdxT, NIDX >, mlx::core::array, mlx::core::fast::CustomKernelShapeInfo, mlx::core::ReductionPlan
• submit() : pocketfft::detail::threading::thread_pool
• Subtract() : mlx::core::Subtract
• Sum : mlx::core::distributed::AllReduce, mlx::core::Reduce, mlx::core::Scan, mlx::core::Scatter, mlx::core::ScatterAxis
• SVD() : mlx::core::SVD
• swizzle() : mlx::steel::BlockSwizzle
• swizzle_log : mlx::steel::GEMMParams, mlx::steel::ImplicitGemmConv2DParams