AtomicVar
976e8babbe
Use compiled black as the pre-commit formatter ( #94 )
2023-12-09 07:06:46 -08:00
Awni Hannun
2520dbcf0a
add losses to the docs, fix black failur ( #92 )
2023-12-09 06:06:52 -08:00
Abe Leininger
430bfb4944
Adds Nesterov momentum to SGD ( #87 )
2023-12-08 23:23:36 -08:00
ShiJZ
08d51bf232
Make it easier to test new optimizers implemented: no need to change test file manually ( #90 )
...
* add helper function get_all_optimizers() in test_optimizers.py
* remove unused import
2023-12-08 21:39:08 -08:00
Kai Ma
cb9e585b8e
Style fix for loss functions ( #91 )
...
* MLE and L1 loss functions
* logsoftmax change and tests
* subtract max logit for numerical stability
* l1 name change
* cross entropy reduction + unit tests
* docstrings
* l1 test name change
* old loss impl + default none
* style
2023-12-08 21:11:56 -08:00
Kai Ma
641d316484
MLE and L1 loss functions ( #88 )
...
* MLE and L1 loss functions
* logsoftmax change and tests
* subtract max logit for numerical stability
* l1 name change
* cross entropy reduction + unit tests
* docstrings
* l1 test name change
* old loss impl + default none
2023-12-08 20:21:37 -08:00
Angelos Katharopoulos
2b714714e1
Add the remainder op ( #85 )
...
* Add remainder in the C++ backend
* Add the python binding and test
2023-12-08 15:08:52 -08:00
Joe Barrow
69a24e6a1e
AdamW implementation ( #72 )
...
* AdamW implementation without bias correction
* Makes use of the underlying Adam implementation
2023-12-08 14:45:34 -08:00
Zach Schillaci
5b9be57ac3
Add isort pre-commit and run ( #68 )
2023-12-08 11:31:47 -08:00
Jagrit Digani
e89c571de7
Update cmake to detect and throw warnings if not on a arm system ( #81 )
...
* Update cmake to detect and throw warnings if not on a arm system
2023-12-08 11:03:25 -08:00
Angelos Katharopoulos
209404239b
Fix the accelerate dispatch for the power op ( #70 )
...
- The exponent and base were swapped because accelerate is using
exponent-base instead of base-exponent
- Fix also the test for binary ops as it was testing op(x, x) which
couldn't catch ordering errors like that
2023-12-08 10:58:03 -08:00
Awni Hannun
4e3bdb560c
random generation fix ( #80 )
...
Random generation fix
2023-12-08 10:40:57 -08:00
Gautam krishna R
86b614afcd
added long_description for pypi readme ( #69 )
2023-12-08 03:33:29 -08:00
Awni Hannun
cfc39d84b7
Some docs on unified memory ( #62 )
...
* doc on unified memory
2023-12-07 19:42:24 -08:00
Zach Schillaci
d11d77e581
Spelling fixes in transformer.py ( #59 )
2023-12-07 13:32:09 -08:00
Jagrit Digani
bf410cb85e
Update CMake to not try and build metallib if Metal framework not found ( #55 )
2023-12-07 09:48:42 -08:00
rushyam
2e126aeb7e
Feature Addition: Encoder-Decoder Transformer Architecture ( #50 )
...
* Implemented decoder-transformer-layer, decoder-transformer and introduce encoder-decoder transformer
* added relu layer
* add src, tgt, memory mask
---------
Co-authored-by: rushyam <rushyam@rushyams-MacBook-Air.local >
2023-12-07 07:37:36 -08:00
Awni Hannun
dfbc52ce56
Install docs + python versions ( #53 )
...
* install + python versions
* add link in install docs
* add link
2023-12-07 07:29:17 -08:00
Angelos Katharopoulos
43e336cff2
Bump the version ( #47 )
2023-12-07 06:40:55 -08:00
Awni Hannun
d895e38f2e
Nits ( #38 )
...
* include 3.12, black format
* circle ci badge
* format
2023-12-06 13:32:41 -08:00
Diogo
d15dead35e
add extra_require with libs for running tests ( #36 )
2023-12-06 12:21:48 -08:00
Jagrit Digani
2440fe0124
NPY loading segfault bug ( #34 )
...
* Fixed Gil semantics in loading and saving from python file streams
2023-12-06 12:03:47 -08:00
Awni Hannun
170e4b2d43
fix links ( #32 )
2023-12-06 08:12:06 -08:00
Jagrit Digani
2629cc8682
Install docs update ( #29 )
...
* Add notes about MacOS version restrictions for mlx in install docs
* Add notes about Xcode version requirements for building from source in install docs
* Let make detect the macosx sdk version being used
* Throw error if trying to build metal kernels with macOS <= 13.4
* Add metal-cpp for macOS 14.2
2023-12-06 08:10:51 -08:00
Ikko Eltociear Ashimine
9f4cf2e0fe
Update extensions.rst ( #26 )
...
unecessary -> unnecessary
2023-12-06 07:18:28 -08:00
Markus Enzweiler
2ffaee0c0d
Updated default argument for stride to 1 in Conv2d() in the docstring ( #22 )
2023-12-06 07:17:58 -08:00
Yingbo Ma
36b245b287
Fix benchmark example ( #11 )
2023-12-06 07:17:16 -08:00
Esakkivel Esakkiraja
8c96b9a890
Update README.md ( #9 )
...
- Fixed typo and other minor errors
2023-12-05 21:31:27 -08:00
Angelos Katharopoulos
07897a346d
Bump the version ( #8 )
...
* Bump the version
* Change the version in the docs as well
2023-12-05 17:46:08 -08:00
Jagrit Digani
d518b3b6a5
Fix gemv broadcasting bug ( #6 )
...
* Fix broadcasting bug in gemv
* Add relevant tests in test_blas.py
2023-12-05 14:15:43 -08:00
Awni Hannun
49cda449b1
apple mlr ( #7 )
2023-12-05 14:10:59 -08:00
Awni Hannun
6449a8682a
Doc theme ( #5 )
...
* change docs theme + links + logo
* move mlx intro to landing page
2023-12-05 12:08:05 -08:00
Angelos Katharopoulos
7546fdb100
Add CircleCI configuration ( #4 )
...
* Add CircleCI configuration
2023-12-04 16:04:11 -08:00
Awni Hannun
db487e6b1a
format
2023-11-30 11:50:50 -08:00
Jagrit Digani
266c4e3df6
Add metal cpp license to acknowledgements ( #3 )
2023-11-30 11:33:23 -08:00
Awni Hannun
46a39e5b1f
copyright + ack
2023-11-30 11:12:53 -08:00
Awni Hannun
a56e0888b8
nits
2023-11-29 16:36:43 -08:00
Awni Hannun
d1926c4752
Readme ( #2 )
...
* readme wip
* more readme
* examples
* spell
* comments + nits
2023-11-29 16:23:42 -08:00
Awni Hannun
adb992a780
contribution and code of conduct ( #1 )
2023-11-29 12:54:28 -08:00
Awni Hannun
8db89dd61d
doc link
2023-11-29 12:45:41 -08:00
Awni Hannun
c1b6bf3f33
missing file
2023-11-29 12:38:32 -08:00
Jagrit Digani
e6306cfee9
jagrit's commit files
2023-11-29 10:52:08 -08:00
Angelos Katharopoulos
d1f86272a2
angelos's commit files
2023-11-29 10:42:59 -08:00
Awni Hannun
8ca7f9e8e9
awni's commit files
2023-11-29 10:30:41 -08:00
Awni Hannun
e411fcae68
Initial commit
2023-11-28 15:33:45 -08:00
