Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[pull] master from arogozhnikov:master #40

Open
wants to merge 316 commits into
base: master
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
316 commits
Select commit Hold shift + click to select a range
5ab5c0d
Update ordering of params in docstring
MilesCranmer Jul 10, 2022
c7451a9
minor
arogozhnikov Jul 15, 2022
e168125
Merge remote-tracking branch 'origin/master' into einsum
Jul 15, 2022
99cb69d
type hints + minor naming + formatting
arogozhnikov Jul 15, 2022
d21031e
remove keras einsum (which is probably wrong)
arogozhnikov Jul 15, 2022
e95f281
minor comment
arogozhnikov Jul 15, 2022
6fbccc1
Clarify documentation re. default bias on EinMix
maxeonyx Jul 26, 2022
697a44c
Merge pull request #201 from maxeonyx/patch-3
arogozhnikov Aug 24, 2022
c0bd89e
corrected spelling mistake: einsops -> einops
cs-mshah Sep 5, 2022
38485ce
Merge pull request #205 from cs-mshah/master
arogozhnikov Sep 6, 2022
ef10e18
add mean-reduction for bfloat16, fix #206
arogozhnikov Sep 27, 2022
b9fe9f9
Merge pull request #209 from arogozhnikov/bfloats
arogozhnikov Sep 27, 2022
32096bc
minor
arogozhnikov Sep 27, 2022
44ad399
minor
arogozhnikov Oct 2, 2022
92aa1d1
change testing framework from nosetests to pytest.
arogozhnikov Oct 2, 2022
de457e2
add py.typed (adopt PEP 561)
arogozhnikov Oct 2, 2022
0756bc6
Merge pull request #211 from arogozhnikov/py-typed
arogozhnikov Oct 2, 2022
b3d0ca2
Delete tensorflow-specific readme
arogozhnikov Oct 2, 2022
4d60e84
Merge pull request #212 from arogozhnikov/remove-tf-readme
arogozhnikov Oct 2, 2022
f8e89e5
change typing of rearrange and parse_shape
arogozhnikov Oct 2, 2022
606f446
reduce number of complaints from mkdocs
arogozhnikov Oct 2, 2022
c7a1727
use hatch and move to pyproject.toml
arogozhnikov Oct 2, 2022
ce301e3
use hatch for pypi deployments (to be tested yet)
arogozhnikov Oct 2, 2022
316bb60
Merge pull request #213 from arogozhnikov/hatch
arogozhnikov Oct 3, 2022
c9b7827
restructure einmix to accomodate for flax.
arogozhnikov Oct 3, 2022
52392b1
adding layers for flax
arogozhnikov Oct 3, 2022
f77170d
add simple one-off testing for flax layers
arogozhnikov Oct 3, 2022
1862e4b
add flax to testing script
arogozhnikov Oct 3, 2022
c56f86a
Merge pull request #214 from arogozhnikov/add-flax-layers
arogozhnikov Oct 3, 2022
1c5dbc4
remark about einsum, flow; delete contributing remarks
arogozhnikov Oct 3, 2022
53c90e6
minor
arogozhnikov Oct 3, 2022
413dd4e
bump version
arogozhnikov Oct 3, 2022
4cd9e16
minor
arogozhnikov Oct 3, 2022
206b31a
minimal typehints in tests
arogozhnikov Oct 10, 2022
7608213
type hinting in backends
arogozhnikov Oct 10, 2022
36c7bb1
type hints + minor changes
arogozhnikov Oct 10, 2022
c054ad0
optimize reductions in torch where possible
arogozhnikov Oct 10, 2022
3ddb01e
type hinting for einmix
arogozhnikov Oct 10, 2022
28ab602
add type casts to layers' implementations to suppress mypy warnings
arogozhnikov Oct 10, 2022
3732e2b
type hints for layers
arogozhnikov Oct 10, 2022
240c7d8
einsum: fix type hints, fix mypy warnigns, more hinting
arogozhnikov Oct 10, 2022
ceb1c4f
fox type hints
arogozhnikov Oct 10, 2022
3823586
add testing as one of hatch scripts
arogozhnikov Oct 10, 2022
5af7e5b
addressing mypy comments
arogozhnikov Oct 10, 2022
4bd1170
use dunders in overloads as positional-only for python 3.7 and earlier
arogozhnikov Oct 10, 2022
1f9f4a7
Update example to match description
EPronovost Oct 16, 2022
b1f905a
Merge pull request #217 from EPronovost/fix-example
arogozhnikov Oct 17, 2022
e2c9bb2
Merge pull request #221 from arogozhnikov/type-hinting
arogozhnikov Nov 5, 2022
5ecfc62
minor formatting things
arogozhnikov Nov 5, 2022
794756e
replace stringy type hint with Any
arogozhnikov Nov 5, 2022
4276550
check in demo for pack and unpack
arogozhnikov Nov 7, 2022
ee13b19
minor
arogozhnikov Nov 7, 2022
048db8b
add readme to docs_src to stop confusing people
arogozhnikov Nov 8, 2022
e65c0fa
undo adding readme because of conflicts with mkdocs's 'strict mode'
arogozhnikov Nov 8, 2022
5a0dbe3
make einsum's docstring compatible with strict mode in mkdocs
arogozhnikov Nov 8, 2022
2c1376f
manually add css for codehilite
arogozhnikov Nov 8, 2022
abf3e6c
minor
arogozhnikov Nov 8, 2022
4bef5f9
minor (formatting, wording)
arogozhnikov Nov 8, 2022
cdfbb71
add concat operation to all backends
arogozhnikov Nov 8, 2022
3dbc79d
bump version to pre, allow top-level import
arogozhnikov Nov 8, 2022
470ed8a
add an implementation of packing
arogozhnikov Nov 8, 2022
3bcaa5c
add tests for packing operations
arogozhnikov Nov 8, 2022
a2012b4
skip some tests for mxnet
arogozhnikov Nov 8, 2022
256458f
resolve pytorch deprecation warning for assert_allclose
arogozhnikov Nov 8, 2022
bc87834
add suppression of warnings for local debugging on mac
arogozhnikov Nov 8, 2022
d5c826d
minor changes in notebook, move to documentation folder
arogozhnikov Nov 8, 2022
d03e487
add note about part 4 of tutorial in readme
arogozhnikov Nov 8, 2022
2da2add
Merge pull request #222 from arogozhnikov/packing
arogozhnikov Nov 8, 2022
e645d22
deabbreviate PS as packed shapes everywhere
arogozhnikov Nov 8, 2022
6aa5733
minor
arogozhnikov Nov 9, 2022
31dc5b3
add softlink to notebook-4 to documentation
arogozhnikov Nov 9, 2022
e305166
remove deprecated requirements
arogozhnikov Nov 9, 2022
ac2e93b
change to an actual version of python
arogozhnikov Nov 9, 2022
3e8252d
add einsum documentation
arogozhnikov Nov 9, 2022
cb3f756
add pack/unpack to documentation/API
arogozhnikov Nov 9, 2022
b85b2ab
include einsum and packing to docs navigation
arogozhnikov Nov 9, 2022
fcd5fe5
exclude index.md in documentation from version control
arogozhnikov Nov 9, 2022
686dc1e
introduce auto-generation of index for documentation
arogozhnikov Nov 9, 2022
b52926c
remove older script for documentation (superseded by hatch commands)
arogozhnikov Nov 9, 2022
5e0bb81
remove video from documentation
arogozhnikov Nov 9, 2022
c1f52fc
minor fixes in tests
arogozhnikov Nov 9, 2022
48da99a
use backend-provided shape
arogozhnikov Nov 9, 2022
c51dcd0
add comment to tests
arogozhnikov Nov 9, 2022
b13332c
check in packing based on data-api
arogozhnikov Nov 9, 2022
ebf084e
Merge pull request #223 from arogozhnikov/packing
arogozhnikov Nov 9, 2022
30ab8ba
multiple updates in readme: mention new functions, add links to the d…
arogozhnikov Nov 9, 2022
506121c
update version to 0.6.0
arogozhnikov Nov 9, 2022
d6f7910
Merge pull request #224 from arogozhnikov/packing
arogozhnikov Nov 9, 2022
fec0613
add a mock setup script
arogozhnikov Dec 4, 2022
1c93652
Update README.md
arogozhnikov Dec 27, 2022
2f96272
update actions to new ones
arogozhnikov Dec 27, 2022
282de2d
remove mxnet backend
arogozhnikov Dec 27, 2022
b49fb75
remove mxnet backend
arogozhnikov Dec 27, 2022
53f205a
do not install mxnet for testing
arogozhnikov Dec 27, 2022
5e3f76a
do not skip oneflow
arogozhnikov Dec 27, 2022
9e36ac6
do not complain on mxnet if not installed
arogozhnikov Dec 27, 2022
7728a07
skip oneflow (see https://github.com/Oneflow-Inc/oneflow/issues/9660)
arogozhnikov Dec 27, 2022
88c1db5
remove mentions of MXNetBackend
arogozhnikov Dec 27, 2022
9336c7a
remove mxnet from function documentation
arogozhnikov Dec 27, 2022
bc54675
minor
arogozhnikov Dec 27, 2022
76e05c3
remove gluon from notebook testing
arogozhnikov Dec 27, 2022
05a8643
Merge pull request #231 from arogozhnikov/remove-mxnet
arogozhnikov Dec 27, 2022
4b1e12a
torch.concat -> torch.cat to support torch < 1.10
arogozhnikov Jan 18, 2023
f569905
Merge pull request #238 from arogozhnikov/torch_cat
arogozhnikov Jan 18, 2023
3c32cfc
Add PaddlePaddle backend for einops
zhwesky2010 Feb 14, 2023
923d91c
fix CI
zhwesky2010 Feb 20, 2023
11664b3
fix CI
zhwesky2010 Feb 21, 2023
eff2833
maintenance: add setting for black formatter
arogozhnikov Mar 15, 2023
7daf074
testing: add infra to select which frameworks are tested
arogozhnikov Mar 15, 2023
0fb7950
flexible setup of frameworks to test
arogozhnikov Mar 15, 2023
a42fc6f
testing: change python version, split frameworks in groups
arogozhnikov Mar 15, 2023
6c5fec5
ci: use strings in yaml
arogozhnikov Mar 15, 2023
fb69804
use typing compatible with py<3.9
arogozhnikov Mar 15, 2023
b7b739a
minor: remove message
arogozhnikov Mar 15, 2023
9fc7ad1
exclude oneflow (fails in py3.10)
arogozhnikov Mar 15, 2023
455140d
no bare torch imports in testing + black formatting
arogozhnikov Mar 15, 2023
be8459b
tests: separate flax and jax in testing
arogozhnikov Mar 15, 2023
66ca360
run dl notebook only with installed backends
arogozhnikov Mar 15, 2023
3026506
testing: one test per notebook
arogozhnikov Mar 15, 2023
6d35671
fix pytorch -> torch
arogozhnikov Mar 15, 2023
275e0ca
minor: typing
arogozhnikov Mar 15, 2023
547ff81
testing: use new method to identify tested frameworks
arogozhnikov Mar 15, 2023
b3a861d
move oneflow to first test group because of protobuf issues
arogozhnikov Mar 15, 2023
a1c704f
massive change in testing
arogozhnikov Mar 15, 2023
90e4075
formatting + reduce pyright's comments on testing
arogozhnikov Mar 15, 2023
8d63968
fix framework name in tests
arogozhnikov Mar 15, 2023
fbc422f
Merge pull request #245 from arogozhnikov/paddle
arogozhnikov Mar 15, 2023
1ea79f8
Autoformat pyproject
arogozhnikov Mar 15, 2023
da5995c
add note about paddle and deprecate gluon
arogozhnikov Mar 15, 2023
52a288a
remove gluon/mxnet specific hacks from testing
arogozhnikov Mar 15, 2023
8f49bd4
gitignore: exclude workspace files
arogozhnikov Mar 15, 2023
2a50316
switch off gluon test
arogozhnikov Mar 15, 2023
b0f2036
remove mxnet-related exemptions from testing
arogozhnikov Mar 15, 2023
534cfd6
minor updates to documentation
arogozhnikov Mar 15, 2023
083b84c
minor
arogozhnikov Mar 16, 2023
a220937
docs: minor changes
arogozhnikov Mar 16, 2023
ad46187
minor: add link to Kapil's video intro
arogozhnikov Mar 16, 2023
32594eb
Merge pull request #246 from arogozhnikov/dev
arogozhnikov Mar 16, 2023
191f7fe
minor: improve css for mkdocs
arogozhnikov Apr 13, 2023
4ce70bc
work on docs styling
arogozhnikov Apr 13, 2023
03f2806
mddocs: change max width of page
arogozhnikov Apr 13, 2023
32fa76e
add allow_ops_in_compiled_graph, and corresponding test
arogozhnikov Apr 17, 2023
1b4b649
add allow_ops_in_compiled_graph, and corresponding test
arogozhnikov Apr 17, 2023
d6bf62f
improve test: comments, randomize input
arogozhnikov Apr 17, 2023
24d14d0
Merge pull request #251 from arogozhnikov/dev
arogozhnikov Apr 19, 2023
18758e8
bump version 0.6.0 -> 0.6.1
arogozhnikov Apr 19, 2023
fa6c083
update readme, update badge to 3.7+
arogozhnikov Apr 19, 2023
103b49d
add .DS_store to gitignore
arogozhnikov Apr 21, 2023
c5f91a9
reformat with black
arogozhnikov Apr 21, 2023
108915d
use f-strings
arogozhnikov Apr 23, 2023
be23380
minor: remove import
arogozhnikov Apr 23, 2023
c5deb93
minor
arogozhnikov May 3, 2023
9727a58
rework recipes and execution logic to preserve axis identity within e…
arogozhnikov May 3, 2023
5e6674c
minor rewrites in redame
arogozhnikov May 12, 2023
6f4c0cc
minor: fix description for einmix
arogozhnikov May 14, 2023
6174175
layers: improve serialization/deserialization
arogozhnikov May 14, 2023
e5863df
drop support for python 3.7
arogozhnikov May 14, 2023
644a804
typos
arogozhnikov May 14, 2023
230ac15
Update README.md
arogozhnikov May 15, 2023
a50db94
Merge pull request #255 from arogozhnikov/dev
arogozhnikov Jul 5, 2023
bcab6ad
change version to 0.6.2rc
arogozhnikov Jul 5, 2023
cec31a8
array api: initial implementation
arogozhnikov Jul 7, 2023
39de675
array api: add tests for rearrange, reduce, repeat based on numpy.arr…
arogozhnikov Jul 7, 2023
92509ac
array api: fix, array api asks to list all dimensions in indexing
arogozhnikov Jul 7, 2023
114ed8e
array api: add tests for array api
arogozhnikov Jul 7, 2023
6cfd017
put incorrect type hint to satisfy torch.jit.script
arogozhnikov Jul 7, 2023
044d195
make distinction in real/fake type hints a bit more clear
arogozhnikov Jul 7, 2023
ae89461
Merge pull request #261 from arogozhnikov/array_api_backend
arogozhnikov Jul 7, 2023
ed4f3c4
type-based caching
arogozhnikov Jul 7, 2023
7274f20
Merge pull request #262 from arogozhnikov/caching
arogozhnikov Jul 7, 2023
488e609
uniform support of list inputs in rearrange/reduce/repeat
arogozhnikov Jul 7, 2023
36f1142
prevent ruff from complaining
arogozhnikov Jul 7, 2023
2bb45fc
add tests for list inputs, use public api for testing repeat
arogozhnikov Jul 7, 2023
b535862
Merge pull request #263 from arogozhnikov/list_inputs
arogozhnikov Jul 7, 2023
6655c69
drop gluon: readme
arogozhnikov Jul 7, 2023
ff52f59
drop gluon + reformat backends.py
arogozhnikov Jul 7, 2023
adc3dd8
dop gluon: remove backend from testing options
arogozhnikov Jul 7, 2023
efb02c4
drop gluon: delete layers
arogozhnikov Jul 7, 2023
f3f31e3
drop gluon: remove tests for gluon layers
arogozhnikov Jul 7, 2023
4c05e8a
drop gluon: remove option in DL notebook
arogozhnikov Jul 7, 2023
1f0cf20
Merge pull request #264 from arogozhnikov/drop_gluon
arogozhnikov Jul 7, 2023
474af4e
automatically register torch ops in torchdynamo
arogozhnikov Jul 8, 2023
ef4028c
Merge pull request #265 from arogozhnikov/autoregister_ops_in_torchdy…
arogozhnikov Jul 8, 2023
39d5358
release 0.7.0rc1, make remarks in README
arogozhnikov Jul 8, 2023
952f066
fix readme rendering in mkdocs
arogozhnikov Jul 8, 2023
f656128
Merge pull request #266 from arogozhnikov/release-070rc1
arogozhnikov Jul 8, 2023
8ac8d4f
introduce fallback if shape was not cachable
arogozhnikov Aug 8, 2023
6fc4e09
Merge pull request #275 from arogozhnikov/torch-dynamic-compile
arogozhnikov Aug 10, 2023
47cbea7
bump version
arogozhnikov Aug 14, 2023
bc89539
Merge pull request #276 from arogozhnikov/torch-dynamic-compile
arogozhnikov Aug 14, 2023
37d7351
Update README.md
arogozhnikov Aug 21, 2023
fb0c43c
fix #279, update description
arogozhnikov Sep 18, 2023
fe91706
use oneflow from pypi for tests
arogozhnikov Sep 18, 2023
3afbfd4
Merge pull request #281 from arogozhnikov/dev
arogozhnikov Sep 18, 2023
fcd36c9
add support for any/all, see #278
arogozhnikov Sep 18, 2023
3f1c740
re-implement bit_count for old pythons (python3.8 in particular)
arogozhnikov Sep 18, 2023
d06ecc0
exclude chainer from tests for any/all
arogozhnikov Sep 18, 2023
a40f793
do not run most tests on any/all, as chainer does not support those
arogozhnikov Sep 18, 2023
e662864
maintenance: cache pip dependencies
arogozhnikov Sep 18, 2023
4f6afb3
Merge pull request #283 from arogozhnikov/pipcache
arogozhnikov Sep 19, 2023
1da562d
maintenance: update versions of github actions
arogozhnikov Sep 19, 2023
c4a0506
Merge pull request #285 from arogozhnikov/update_actions
arogozhnikov Sep 25, 2023
f65eeed
set version to 0.7.0
arogozhnikov Oct 1, 2023
5b96062
Merge remote-tracking branch 'origin/master'
arogozhnikov Oct 1, 2023
aa3270a
minor: correct readme
arogozhnikov Oct 1, 2023
af77dc6
projects: add a link to einops-cpp
arogozhnikov Oct 2, 2023
e61294d
Coerce bool to int
drubinstein Oct 5, 2023
c97aac7
tests: fix torch installation to force CPU
arogozhnikov Oct 5, 2023
6b7628f
tests: fix jax installation in CI, force CPU
arogozhnikov Oct 5, 2023
ed0dca0
tests: accompany each import fail with a name of framework that cause…
arogozhnikov Oct 5, 2023
db1501b
tests: drop oneflow
arogozhnikov Oct 6, 2023
8811af4
Merge pull request #289 from arogozhnikov/remove-oneflow-from-testing
arogozhnikov Oct 6, 2023
e3082f2
Merge pull request #287 from drubinstein/patch-1
arogozhnikov Oct 6, 2023
e240725
remove affiliations (changed)
arogozhnikov Oct 11, 2023
b65eb59
Merge remote-tracking branch 'origin/master'
arogozhnikov Oct 11, 2023
a6e9353
update link to einops-cpp
arogozhnikov Oct 11, 2023
970ed5a
add tinygrad, or at least the straightforward part
blueridanus Dec 7, 2023
db815ca
Fix invalid escape sequence in einsum docstring
atwam Dec 8, 2023
194dad8
einsum support
blueridanus Dec 9, 2023
44080db
Merge pull request #298 from atwam/patch-1
arogozhnikov Dec 11, 2023
a2ab65c
add announcement
arogozhnikov Dec 21, 2023
d0c7fee
Merge remote-tracking branch 'origin/master'
arogozhnikov Dec 21, 2023
bb6ce2c
Merge pull request #297 from blueridanus/backend/tinygrad
arogozhnikov Jan 3, 2024
d495e7c
minimize announcement
arogozhnikov Jan 8, 2024
9a9b304
delete experimental data-api_packing as production version is availab…
arogozhnikov Jan 10, 2024
5b41d90
allow anonymous axes in parse_shape, fix #302
arogozhnikov Jan 11, 2024
5655ce1
Merge pull request #303 from arogozhnikov/dev
arogozhnikov Jan 11, 2024
60b87de
remove announcement
arogozhnikov Feb 11, 2024
2ab5c2a
switch to stable paddlepaddle in testing, because of https://github.c…
arogozhnikov Apr 27, 2024
cff63f2
Codebase standards + update TF layers (#318)
arogozhnikov Apr 28, 2024
f33113e
update github actions
arogozhnikov Apr 28, 2024
b67cfea
Rename KerasBackend -> TFKerasBackend to avoid confusion with keras 3.
arogozhnikov Apr 28, 2024
10e19b4
prepare readme for a new release
arogozhnikov Apr 28, 2024
fe9d81d
bump version to 0.8.0
arogozhnikov Apr 28, 2024
0ea28c8
Update README.md
project-delphi Apr 18, 2024
6181e1e
Update README.md
arogozhnikov May 26, 2024
35f74b3
mention that MLX supports array api standard
arogozhnikov Aug 7, 2024
530d17a
exclude chainer from CI
arogozhnikov Aug 7, 2024
11680b4
fix tests for numpy regression (see https://github.com/numpy/numpy/is…
arogozhnikov Aug 8, 2024
ef4cd25
skip array api tests for numpy v1 (installed by paddle during tests)
arogozhnikov Aug 8, 2024
309337d
switch off paddle in CI as well
arogozhnikov Aug 8, 2024
4eea312
fix concat in tinygrad backend
pzread Sep 13, 2024
4db03a2
do not reformat notebooks
arogozhnikov Sep 13, 2024
10dbd8d
do not reformat notebooks (CI kept formatting despite previous config…
arogozhnikov Sep 13, 2024
88719b2
add devcontainer and dockerfile
arogozhnikov Sep 14, 2024
bbb434b
remove deprecated testing command, prefer to use the same venv + devc…
arogozhnikov Sep 14, 2024
1a078a6
move part of installation to devcontainer
arogozhnikov Sep 14, 2024
c642d16
lxml[html_clean] is needed because of some recent changes in lxml(?)
arogozhnikov Sep 14, 2024
f5a009f
delete experimental/indexing
arogozhnikov Sep 15, 2024
b33f7e6
Use same ruff version in CI and in pre-commit (#334)
arogozhnikov Sep 15, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
switch to stable paddlepaddle in testing, because of PaddlePaddle/Pad…
  • Loading branch information
arogozhnikov committed Apr 27, 2024
commit 2ab5c2a5f693648a16eb3ceb8d2e1ed857101bde
4 changes: 3 additions & 1 deletion test.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,9 @@ def main():
"tensorflow": ["tensorflow"],
"chainer": ["chainer"],
"cupy": ["cupy"],
"paddle": ["paddlepaddle==0.0.0 -f https://www.paddlepaddle.org.cn/whl/linux/cpu-mkl/develop.html"],
# switch to stable paddlepaddle, because of https://github.com/PaddlePaddle/Paddle/issues/63927
# "paddle": ["paddlepaddle==0.0.0 -f https://www.paddlepaddle.org.cn/whl/linux/cpu-mkl/develop.html"],
"paddle": ["paddlepaddle"],
"oneflow": ["oneflow==0.9.0"],
}

Expand Down