List of all items
Structs
- Array
- Complex64
- CustomCall
- CustomFunction
- Device
- Flags
- Fused
- MappedFile
- Scheduled
- SliceArg
- Stream
- bf16
- f16
- fast::KernelCall
- fast::KernelConfig
- fast::MetalKernel
- fast::Rotation
Enums
- ArgReduceOp
- BinaryOp
- Category
- DType
- DeviceKind
- Error
- FusedRef
- Primitive
- ReduceOp
- Scalar
- ScalarOrArray
- ScanOp
- ScatterReduce
- UnaryOp
- fast::AttentionMask
- fast::FastOp
- fast::TemplateArg
Traits
Functions
- abs
- add
- addmm
- all
- any
- arange
- argmax
- argmin
- argpartition
- argsort
- astype
- async_eval
- at_least_float
- broadcast_to
- checkpoint
- concatenate
- conjugate
- contiguous
- conv1d
- cos
- cummax
- cummin
- cumprod
- cumsum
- custom_function
- default_device
- default_stream
- dequantize
- divide
- equal
- erfinv
- eval
- exp
- expand_dims
- fast::add_rms_norm
- fast::cross_entropy
- fast::gated_delta
- fast::gated_delta_steps
- fast::layer_norm
- fast::norm_rope
- fast::rms_norm
- fast::rope
- fast::scaled_dot_product_attention
- gather_mm
- gather_qmm
- grad
- greater
- greater_equal
- jvp
- less
- less_equal
- log
- logaddexp
- logcumsumexp
- logsumexp
- matmul
- max
- maximum
- mean
- min
- minimum
- multiply
- negative
- not_equal
- ones
- partition
- prod
- promote_types
- put_along_axis
- quantize
- quantized_matmul
- random::bernoulli
- random::bits
- random::categorical
- random::gumbel
- random::key
- random::next_key
- random::normal
- random::randint
- random::seed
- random::split
- random::split_pair
- random::uniform
- repeat
- reshape
- scatter_add_along_axis
- schedule
- set_default_stream
- sigmoid
- sign
- sin
- slice
- slice_update
- softmax
- sort
- split
- split_at
- sqrt
- squeeze
- stack
- stop_gradient
- subtract
- sum
- take
- take_along_axis
- tanh
- topk
- transpose
- value_and_grad
- vjp
- vmap
- where_
- zeros