Commit Graph

  • fffe072028 Implementation of mlx.random.multivariate_normal (#502) (#877) Luca Arnaboldi 2024-04-09 22:50:12 +02:00
  • a1a31eed27 Add mx.meshgrid (#961) Abe Leininger 2024-04-09 14:43:08 -04:00
  • ae812350f9 use string (#976) Awni Hannun 2024-04-09 11:22:00 -07:00
  • b63ef10a7f Extensions (#962) Awni Hannun 2024-04-09 08:50:36 -07:00
  • 42afe27e12 std and expm1 (#973) Awni Hannun 2024-04-08 14:26:01 -07:00
  • 76e63212ff Enable bfloat scan (#974) Awni Hannun 2024-04-08 12:29:19 -07:00
  • aac2f9fb61 Improve profiling with gpu tracing (#969) Awni Hannun 2024-04-07 21:47:43 -07:00
  • bddf23f175 patch bump (#956) v0.9.1 Awni Hannun 2024-04-04 11:56:37 -07:00
  • 039da779d1 No quant reshape (#957) Awni Hannun 2024-04-04 11:52:12 -07:00
  • d88d2124b5 segfaut layer norm grad (#955) Awni Hannun 2024-04-04 10:59:15 -07:00
  • e142aaf8a1 Option for precise softmax (#953) Awni Hannun 2024-04-04 08:32:35 -07:00
  • 0caf35f4b8 Better exceptions in case of invalid operations on mlx.core.array (#910) (#926) AmirHossein_Razlighi 2024-04-03 07:41:24 +03:30
  • 3fc993f82d Properly handle negative axes in python vmap (#944) Angelos Katharopoulos 2024-04-02 18:07:23 -07:00
  • 741eb28443 fix a couple bugs (#952) Awni Hannun 2024-04-02 12:07:41 -07:00
  • 1a87dc5ea8 Fix compile fusion for multi-output edge cases (#950) Angelos Katharopoulos 2024-04-02 08:42:31 -07:00
  • 2427fa171e Fix cpu compile (#934) Awni Hannun 2024-04-01 17:37:12 -07:00
  • 639e06e1f3 Indexing bug fix (#947) Jagrit Digani 2024-04-01 12:18:50 -07:00
  • 02fedbf1da Fix array initialization from list (#942) Angelos Katharopoulos 2024-04-01 06:27:52 -07:00
  • 110d9b149d Layer norm grad fix donation bug (#941) Angelos Katharopoulos 2024-04-01 06:15:50 -07:00
  • 9cbff5ec1d Fix typo in qmm check (#940) Angelos Katharopoulos 2024-03-31 19:15:44 -07:00
  • 433c0206b0 Update saving_and_loading.rst (#929) Suvan Kumar 2024-03-30 14:30:06 -07:00
  • 8915901966 Donation bug (#933) Awni Hannun 2024-03-30 10:08:54 -07:00
  • f48bc496c7 Comparing python objects (such as list/tuple) with mlx.core.array (#920) AmirHossein_Razlighi 2024-03-29 17:22:30 +03:30
  • 913b19329c Add missing && when forwarding args (#925) Cheng 2024-03-29 22:48:29 +09:00
  • d8cb3128f6 bump (#924) v0.9.0 Awni Hannun 2024-03-28 16:14:55 -07:00
  • 5f9ba3019f Fix qmm_t for unaligned cases (#923) Angelos Katharopoulos 2024-03-28 15:34:57 -07:00
  • 46caf0bef0 Remove unnecessary string copies (#891) Cheng 2024-03-29 05:14:59 +09:00
  • 45f636e759 Add Metal debug option and capture functions (#707) Jack Mousseau 2024-03-28 09:40:31 -07:00
  • a7b404ff53 Use uintptr_t instead of size_t to store funtion id (#916) Cheng 2024-03-28 22:37:59 +09:00
  • c4fd0e5ede Fixes #918 bug in compile_tests (#919) Angelos Katharopoulos 2024-03-27 22:37:37 -07:00
  • bab5386306 Make ops aware of rvalues: astype/as_strided/copy/full (#895) Cheng 2024-03-28 14:35:55 +09:00
  • aca7584635 Fix OOB read in qmv when non-divisible by blocksize (#917) Angelos Katharopoulos 2024-03-27 22:18:35 -07:00
  • d611251502 Support Chaining for some of functionalities of nn.Module (#885) (#897) AmirHossein_Razlighi 2024-03-28 06:28:29 +03:30
  • f30b659291 Make MLX build on x64 macOS (#901) Cheng 2024-03-27 22:14:29 +09:00
  • 90dfa43ff1 Don't use make_unique to create shared_ptr (#902) Cheng 2024-03-27 22:13:29 +09:00
  • dc175f08d3 Fix race in multi-stream eval (#911) Awni Hannun 2024-03-26 16:36:36 -07:00
  • 29221fa238 Implement vjps for some primitives in the fast namespace (#883) Angelos Katharopoulos 2024-03-26 16:35:34 -07:00
  • a789685c63 Remove duplicate defines of StreamOrDevice and is_big_endian (#892) Cheng 2024-03-27 07:15:11 +09:00
  • 240d10699c Implement negative padding in conv with slicing (#907) Jagrit Digani 2024-03-26 14:59:19 -07:00
  • 925014b661 Fix multiblock sort limits (#906) Jagrit Digani 2024-03-26 14:00:00 -07:00
  • 5611e1a95e Fix unsqueeze with None (#899) Abdussamet Türker 2024-03-26 23:59:44 +03:00
  • 570f2bf29e pick up preivously set attributes (#905) Awni Hannun 2024-03-26 11:19:59 -07:00
  • 9948eddf11 Fix nan and improve speed for qvm (#903) Angelos Katharopoulos 2024-03-26 10:41:45 -07:00
  • a3ee03da01 Fixing random.normal for half-precision dtype #642 (#904) Luca Arnaboldi 2024-03-26 17:58:27 +01:00
  • 28fcd2b519 Add missing && when forwarding args (#894) Cheng 2024-03-26 06:55:54 +09:00
  • 8e686764ac Ensure shape dimensions are within supported integer range (#566) (#704) Jack Mousseau 2024-03-25 13:29:45 -07:00
  • 479051ce1c add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) Daniel Strobusch 2024-03-25 20:32:59 +01:00
  • bfb5bad4f0 patch (#893) v0.8.1 Awni Hannun 2024-03-24 21:03:59 -07:00
  • 1e16331d9c post nanobind docs fixes and some updates (#889) Awni Hannun 2024-03-24 15:03:27 -07:00
  • be98f4ab6b Reduce a little overhead (#871) Awni Hannun 2024-03-22 17:29:36 -07:00
  • 6ee1112f30 Fix copy donation and add partial rope (#881) Angelos Katharopoulos 2024-03-22 17:28:26 -07:00
  • 8e5a5a1ccd Set item bug fix (#879) Jagrit Digani 2024-03-22 12:11:17 -07:00
  • fcda3a0e66 Increase test tolerance for fast.layer_norm (#880) Angelos Katharopoulos 2024-03-22 12:10:27 -07:00
  • 9663c22fe9 Do not store iostream in shared_ptr (#872) Cheng 2024-03-22 22:54:45 +09:00
  • f0ae00da12 Reduce implicit copies in make_array (#874) Cheng 2024-03-22 22:29:16 +09:00
  • 44390bd3d0 Bump (#869) v0.8.0 Awni Hannun 2024-03-21 13:56:56 -07:00
  • 2225374060 Adds mx.fast.layer_norm (#870) Angelos Katharopoulos 2024-03-21 13:55:51 -07:00
  • 105d236889 Add vmap for SVD and inverse (#849) nicolov 2024-03-21 21:18:27 +01:00
  • 53e6a9367c Use reshape and transpose for non-overlapping pooling windows (#867) Angelos Katharopoulos 2024-03-21 10:21:03 -07:00
  • f5a1582fe8 Add minimum for cosine decay function (#859) Chime Ogbuji 2024-03-21 10:33:29 -04:00
  • a54f06b16f Fast RMS Norm (#862) Awni Hannun 2024-03-21 07:20:54 -07:00
  • 4650d94d98 Add missing && in eval (#864) Cheng 2024-03-21 22:15:48 +09:00
  • a5681ebc52 Update set item (#861) Jagrit Digani 2024-03-21 02:48:13 -07:00
  • e849b3424a Do not use static constexpr in header (#863) Cheng 2024-03-21 13:28:05 +09:00
  • b219d12a6b Check edge case handling in row reduce med kernel (#858) Jagrit Digani 2024-03-20 11:37:58 -07:00
  • cec8661113 Add a SliceUpdate op and primitive (#850) Jagrit Digani 2024-03-20 10:39:25 -07:00
  • 73a8c090e0 Pass shape and inputs by value in array's constructor (#853) Cheng 2024-03-20 23:54:30 +09:00
  • db6796ac61 simple typo fille (#848) Md. Rasel Mandol 2024-03-19 18:45:17 +05:30
  • 9a8ee00246 Switch to nanobind (#839) Awni Hannun 2024-03-18 20:12:25 -07:00
  • d39ed54f8e Some C++ code are not needed (#841) Cheng 2024-03-19 09:04:10 +09:00
  • 16546c70d8 No reshape rope (#838) Awni Hannun 2024-03-18 17:03:07 -07:00
  • eaba55c9bf Add matrix inversion primitive (#822) nicolov 2024-03-15 14:34:36 +01:00
  • 19ec023256 vmap matmul and admm (#836) Awni Hannun 2024-03-14 14:38:22 -07:00
  • 63ab0ab580 version (#835) v0.7.0 Awni Hannun 2024-03-14 12:20:40 -07:00
  • 8dfc376c00 Strided reduce specialization for small reductions (#826) Jagrit Digani 2024-03-14 09:16:53 -07:00
  • 1efee9db09 Add types and order in kernel name (#831) Angelos Katharopoulos 2024-03-13 20:34:06 -07:00
  • 43abc402d8 route to fallback (#828) Awni Hannun 2024-03-13 19:56:04 -07:00
  • 3f8b1668c4 Make reshape faster for row_contiguous cases (#829) Angelos Katharopoulos 2024-03-13 16:22:03 -07:00
  • 76c919b4ec NumberOfElements for shapeless compile and vmap fixes (#802) Angelos Katharopoulos 2024-03-13 10:34:14 -07:00
  • 29d0c10ee5 Reshape improvement (#818) Angelos Katharopoulos 2024-03-12 17:54:31 -07:00
  • 5ad133f8bb No copy gems (#801) Jagrit Digani 2024-03-12 13:13:41 -07:00
  • d0c544a868 Add SVD primitive (#809) nicolov 2024-03-12 20:30:11 +01:00
  • ffb19df3c0 Fix docstring for correctly rendering (#820) Daniel Falbel 2024-03-12 15:46:44 -03:00
  • 8b7532b9ab fix scatter (#821) Awni Hannun 2024-03-12 11:42:07 -07:00
  • 366478c560 fix modules with dict (#819) Awni Hannun 2024-03-12 08:54:06 -07:00
  • 8e5600022a Implement RNN, GRU, LSTM (#268) Justin Deschenaux 2024-03-12 05:14:44 +01:00
  • 0e95b64942 Fix bug in tape order during simplify (#816) Awni Hannun 2024-03-11 17:29:05 -07:00
  • 0ae22b915b Remove code duplication in reduce ops (#793) nicolov 2024-03-11 18:57:07 +01:00
  • 7c441600fe Compile stride bug (#812) Awni Hannun 2024-03-11 06:31:31 -07:00
  • a4d290adb9 Remove depth traversal (#813) Awni Hannun 2024-03-09 20:21:32 -08:00
  • 28301807c2 Version bump and os error (#807) v0.6.0 Awni Hannun 2024-03-07 13:57:58 -08:00
  • 74ed0974b3 Support 13.0+ with xcode 14.3 (#806) Awni Hannun 2024-03-07 13:27:57 -08:00
  • ec8a4864fa Fix SDPA kernel bug on Mac OS 13.3 SDK (#805) Jagrit Digani 2024-03-07 10:18:09 -08:00
  • b7588fd5d7 fix inplace to not make a shallow copy (#804) Awni Hannun 2024-03-07 09:34:11 -08:00
  • f512b905c7 Minimum xcode / sdk (#800) Awni Hannun 2024-03-07 08:19:43 -08:00
  • afd5274049 route to fallback for bfloat (#794) Awni Hannun 2024-03-06 15:39:12 -08:00
  • 1074674e32 Add a maximum graph depth (#797) Awni Hannun 2024-03-06 15:39:00 -08:00
  • 7762e07fde Update function_transforms.rst (#796) AlexCheema 2024-03-06 20:03:37 +00:00
  • cbefd9129e Implementation of pickle, copy and deepcopy for Python arrays (#300 & #367). (#713) Luca Arnaboldi 2024-03-06 17:02:41 +01:00
  • e39bebe13e Fix reshaping of empty arrays (#791) Angelos Katharopoulos 2024-03-05 23:33:22 -08:00