Modulesยง
- apply_
rope - binary
- broadcast
- cast
- change_
axes - concat
- copy_
based - Translators for ops that only need the generic copy_nd dispatch. These are fully backend-agnostic and can be constructed without any backend-specific arguments.
- dyn_
kv_ cache - element_
wise - gelu_
approximate - iff
- leaky_
relu - pulse
- reduce
- rms_
norm - rotate_
half - scaled_
masked_ softmax - slice
- softmax