- abs
- abs_
- add
- add_
- addmm
- all
- allclose
- amax
- angle
- any
- apply_rotary_pos_emb
- arange
- argmax
- argmin
- batch_norm
- bitwise_and
- bitwise_and_
- bitwise_not
- bitwise_not_
- bitwise_or
- bitwise_or_
- bmm
- cat
- clamp
- clamp_
- concat_and_cache_mla
- contiguous
- conv1d
- conv2d
- conv3d
- cos
- cos_
- count_nonzero
- cross_entropy_loss
- cummax
- cummin
- cumsum
- diag
- diag_embed
- diagonal
- div
- div_
- dot
- dropout
- elu
- embedding
- eq
- erf
- erf_
- exp
- exp_
- exponential_
- eye
- fill
- fill_
- flash_attention_forward
- flash_attn_varlen_func
- flash_mla
- flip
- floor_divide
- floor_divide_
- full
- full_like
- fused_add_rms_norm
- gather
- ge
- gelu
- gelu_
- gelu_and_mul
- glu
- group_norm
- gt
- hstack
- index
- index_add
- index_put
- index_put_
- index_select
- instance_norm
- isclose
- isfinite
- isin
- isinf
- isnan
- kron
- layer_norm
- le
- lerp
- lerp_
- linspace
- log
- log_sigmoid
- log_softmax
- logical_and
- logical_not
- logical_or
- logical_xor
- lt
- masked_fill
- masked_fill_
- masked_select
- max
- maximum
- mean
- min
- minimum
- mm
- mse_loss
- mul
- mul_
- multinomial
- mv
- nan_to_num
- ne
- neg
- neg_
- nll_loss
- nonzero
- normal
- ones
- ones_like
- outer
- pad
- polar
- pow
- pow_
- prod
- quantile
- rand
- rand_like
- randn
- randn_like
- randperm
- reciprocal
- reciprocal_
- relu
- relu_
- remainder
- remainder_
- repeat
- repeat_interleave
- reshape_and_cache
- reshape_and_cache_flash
- resolve_conj
- resolve_neg
- rms_norm
- rsqrt
- rsqrt_
- scaled_dot_product_attention
- scatter
- scatter_
- select_scatter
- sigmoid
- sigmoid_
- silu
- silu_
- silu_and_mul
- sin
- sin_
- skip_layer_norm
- slice_scatter
- softmax
- sort
- sqrt
- stack
- sub
- sub_
- sum
- tanh
- tanh_
- threshold
- tile
- to
- topk
- triu
- uniform_
- unique
- upsample
- var_mean
- vdot
- vector_norm
- vstack
- weight_norm
- where
- zeros
- zeros_like