zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
8bd6bfa4b5
mlx/docs/src/python/ops.rst

153 lines
1.6 KiB
ReStructuredText
Raw Normal View History

angelos's commit files
2023-11-30 02:42:59 +08:00
.. _ops:
Operations
==========
.. currentmodule:: mlx.core
std and expm1 (#973) * std and expm1 * actually add expm1 * fix linux * fix vjp * relax tol for linux test * Add it to the compilable primitives --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-04-09 05:26:01 +08:00
.. autosummary::
angelos's commit files
2023-11-30 02:42:59 +08:00
:toctree: _autosummary
abs
add
all
std and expm1 (#973) * std and expm1 * actually add expm1 * fix linux * fix vjp * relax tol for linux test * Add it to the compilable primitives --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-04-09 05:26:01 +08:00
allclose
angelos's commit files
2023-11-30 02:42:59 +08:00
any
arange
arccos
arccosh
arcsin
arcsinh
arctan
Added ArcTan2 operation (#1079) * Added ArcTan2 operation * Cleanup, bug fixes from code review * Minor cleanup, fixed Linux tests
2024-05-08 23:35:15 +08:00
arctan2
angelos's commit files
2023-11-30 02:42:59 +08:00
arctanh
argmax
argmin
argpartition
argsort
array_equal
Added support for atleast_1d, atleast_2d, atleast_3d (#694)
2024-02-20 01:40:52 +08:00
atleast_1d
atleast_2d
atleast_3d
Add bitwise ops (#1037) * bitwise ops * fix tests
2024-04-27 13:03:42 +08:00
bitwise_and
bitwise_or
bitwise_xor
Masked mm (#978) * Add block masked matmul op and primitive
2024-04-17 05:45:39 +08:00
block_masked_mm
Block sparse mm (#1058)
2024-05-03 05:03:58 +08:00
block_sparse_mm
Add bitwise ops (#1037) * bitwise ops * fix tests
2024-04-27 13:03:42 +08:00
broadcast_to
Floor and Ceil (#150) * Implements Floor and Ceil Ops
2023-12-15 02:00:23 +08:00
ceil
Added clip function (#159) * Added clip * Added Python bindings * Formatting * Added cpp tests * Added Python tests * python bindings work * rebase --------- Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-18 12:00:29 +08:00
clip
angelos's commit files
2023-11-30 02:42:59 +08:00
concatenate
convolve
conv1d
conv2d
Convolution update (#651) * Init steel conv and update Conv primitive * Update slow CPU implementation to support flipping and input dilation winograd conv routing Co-authored-by: Awni Hannun <awni@apple.com>
2024-02-29 12:11:16 +08:00
conv_general
angelos's commit files
2023-11-30 02:42:59 +08:00
cos
cosh
post nanobind docs fixes and some updates (#889) * post nanobind docs fixes and some updates * one more doc nit * fix for stubs and latex
2024-03-25 06:03:27 +08:00
cummax
cummin
cumprod
cumsum
Adds radians and degrees (#1011)
2024-04-23 02:17:49 +08:00
degrees
Adds C++ and nn quantization utilities (#230) * Add C++ de-/quantize ops * Add quantize functions to the docs and tests * Add a QuantizedLinear module
2023-12-21 06:17:38 +08:00
dequantize
Implement diagonal operator (#562) * Implement diagonal operator This implements mx.diagonal in operator level, inspired by @ManishAradwad. * added `mx.diag` with tests * corrected few things * nits in bindings * updates to diag --------- Co-authored-by: ManishAradwad <manisharadwad@gmail.com> Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-31 01:45:48 +08:00
diag
diagonal
angelos's commit files
2023-11-30 02:42:59 +08:00
divide
Multi output primitives (#330) * Multi-output primitives --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-01-09 08:39:08 +08:00
divmod
angelos's commit files
2023-11-30 02:42:59 +08:00
equal
erf
erfinv
exp
std and expm1 (#973) * std and expm1 * actually add expm1 * fix linux * fix vjp * relax tol for linux test * Add it to the compilable primitives --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-04-09 05:26:01 +08:00
expm1
angelos's commit files
2023-11-30 02:42:59 +08:00
expand_dims
Added eye/identity ops (#119) `eye` and `identity` C++ and Python ops
2023-12-12 04:38:17 +08:00
eye
implemented Flatten Module (#149) * implemented flatten op --------- Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-17 13:54:37 +08:00
flatten
Implements divide for integer types and adds floor_divide op (#228) * Add floor_divide * Add floor_divide to the tests * Add floor_divide to the docs
2023-12-20 12:12:19 +08:00
floor
floor_divide
angelos's commit files
2023-11-30 02:42:59 +08:00
full
greater
greater_equal
Added eye/identity ops (#119) `eye` and `identity` C++ and Python ops
2023-12-12 04:38:17 +08:00
identity
Add inner / outer op (#348) * inner / outer impl * python tests * ops list and ack * updated descriptions * use test helper * removed dtype check and flatten outer to 1-D * updated docs * just use the reshape to flatten
2024-01-08 01:01:09 +08:00
inner
bindings for memory info (#761) * bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error
2024-03-02 11:51:58 +08:00
isclose
add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) * add numeric type hierarchy and issubdtype as well as a set_dtype method to nn.Module with predicate numeric type hierarchy and issubtype is compatible to the [numpy hierarchy](https://github.com/numpy/numpy/blob/220f0ab2c5958f9d49b9a28c69a760657b06d9b9/numpy/_core/numerictypes.py#L42). Closes #285. * nits in docs * unify type category checking * nits in docs * nits in docs * more docs nits * fix callable type --------- Co-authored-by: Awni Hannun <awni@apple.com>
2024-03-26 03:32:59 +08:00
isinf
Add isnan (#423)
2024-01-13 03:16:48 +08:00
isnan
implemented isposinf and isneginf in one PR (#470) * ran precommit * updated docs
2024-01-16 22:48:07 +08:00
isneginf
add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) * add numeric type hierarchy and issubdtype as well as a set_dtype method to nn.Module with predicate numeric type hierarchy and issubtype is compatible to the [numpy hierarchy](https://github.com/numpy/numpy/blob/220f0ab2c5958f9d49b9a28c69a760657b06d9b9/numpy/_core/numerictypes.py#L42). Closes #285. * nits in docs * unify type category checking * nits in docs * nits in docs * more docs nits * fix callable type --------- Co-authored-by: Awni Hannun <awni@apple.com>
2024-03-26 03:32:59 +08:00
isposinf
Add bitwise ops (#1037) * bitwise ops * fix tests
2024-04-27 13:03:42 +08:00
left_shift
angelos's commit files
2023-11-30 02:42:59 +08:00
less
less_equal
Added linspace (#181) * linspace ops support --------- Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-19 11:57:55 +08:00
linspace
angelos's commit files
2023-11-30 02:42:59 +08:00
load
log
log2
log10
log1p
logaddexp
logical_not
feat: add logicalAnd and logicalOR (#386) * feat: add logicalAnd and logicalOR * run pre-commit * Refactor logical_and and logical_or functions * Add acknowledgement * Add logical AND and logical OR operators * Refactor logical_and and logical_or functions * Add support for logical operators on bool arrays * Update mlx/ops.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update mlx/ops.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Add logical AND and OR operators for arrays and scalars * Refactor vjp and jvp methods in primitives.cpp * Add overloaded operators for logical AND and OR * format --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com> Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-08 23:00:05 +08:00
logical_and
logical_or
angelos's commit files
2023-11-30 02:42:59 +08:00
logsumexp
matmul
max
maximum
mean
Add mx.meshgrid (#961)
2024-04-10 02:43:08 +08:00
meshgrid
angelos's commit files
2023-11-30 02:42:59 +08:00
min
minimum
Add move and swap axis, and vmap for slice, concat, and gather (#158) * add move and swap axis, and vmap for slice, concat, and gather
2023-12-15 04:59:12 +08:00
moveaxis
angelos's commit files
2023-11-30 02:42:59 +08:00
multiply
negative
Expose function to clear memory cache (#1032) * expose function to clear memory cache * fix linux build * fix metal tests
2024-04-25 07:48:51 +08:00
not_equal
angelos's commit files
2023-11-30 02:42:59 +08:00
ones
ones_like
Add inner / outer op (#348) * inner / outer impl * python tests * ops list and ack * updated descriptions * use test helper * removed dtype check and flatten outer to 1-D * updated docs * just use the reshape to flatten
2024-01-08 01:01:09 +08:00
outer
angelos's commit files
2023-11-30 02:42:59 +08:00
partition
pad
prod
Adds C++ and nn quantization utilities (#230) * Add C++ de-/quantize ops * Add quantize functions to the docs and tests * Add a QuantizedLinear module
2023-12-21 06:17:38 +08:00
quantize
quantized_matmul
Adds radians and degrees (#1011)
2024-04-23 02:17:49 +08:00
radians
angelos's commit files
2023-11-30 02:42:59 +08:00
reciprocal
Add support for repeat (#278) * add repeat function * fix styling * optimizing repeat * fixed minor issues * not sure why that folder is there xD * fixed now for sure * test repeat not repeat test * Fixed --------- Co-authored-by: Bahaa Eddin tabbakha <bahaa@Bahaas-MacBook-Pro.local>
2023-12-28 05:11:38 +08:00
repeat
angelos's commit files
2023-11-30 02:42:59 +08:00
reshape
Add bitwise ops (#1037) * bitwise ops * fix tests
2024-04-27 13:03:42 +08:00
right_shift
Adds round op and primitive (#203)
2023-12-19 03:32:48 +08:00
round
angelos's commit files
2023-11-30 02:42:59 +08:00
rsqrt
save
savez
savez_compressed
GGUF support (#350) * Initial GGUF support for tensor fields. --------- Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-11 05:22:48 +08:00
save_gguf
Safetensor support (#215) Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-27 18:06:55 +08:00
save_safetensors
angelos's commit files
2023-11-30 02:42:59 +08:00
sigmoid
sign
sin
sinh
softmax
sort
split
sqrt
square
squeeze
Added mx.stack c++ frontend impl (#123) * stack C++ operation + python bindings
2023-12-15 05:21:19 +08:00
stack
std and expm1 (#973) * std and expm1 * actually add expm1 * fix linux * fix vjp * relax tol for linux test * Add it to the compilable primitives --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-04-09 05:26:01 +08:00
std
angelos's commit files
2023-11-30 02:42:59 +08:00
stop_gradient
subtract
sum
Add move and swap axis, and vmap for slice, concat, and gather (#158) * add move and swap axis, and vmap for slice, concat, and gather
2023-12-15 04:59:12 +08:00
swapaxes
angelos's commit files
2023-11-30 02:42:59 +08:00
take
take_along_axis
tan
tanh
Add Tensordot op (#344)
2024-01-03 09:15:00 +08:00
tensordot
bindings for memory info (#761) * bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error
2024-03-02 11:51:58 +08:00
tile
Pickle + dtype fix for numpy conversion (#763) * pickle + dtype fix for numpy conversion * fix getattribute on Module base * remove unused function * fix tests * add topk to ops * fix doc
2024-03-02 22:09:29 +08:00
topk
angelos's commit files
2023-11-30 02:42:59 +08:00
transpose
added tri / tril / triu (#170) * added tri / tril / triu * fixed tests * ctest tests * tri overload and simplified tests * changes from comment * more tests for m * ensure assert if not 2-D * remove broadcast_to * minor tweaks --------- Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-16 09:30:34 +08:00
tri
tril
triu
angelos's commit files
2023-11-30 02:42:59 +08:00
var
where
zeros
zeros_like
Reference in New Issue Copy Permalink