Comments (4)
arogozhnikov
commented on May 21, 2024
1
@michaelklachko
it is a good example for using einsum, probably worth adding to a collection of 'writing better code'
from einops.
arogozhnikov
commented on May 21, 2024
@michaelklachko done:
https://github.com/arogozhnikov/readable_capsnet
Capsule networks in 75 lines (for encoder/decoder + separate stackable layer for routing).
Now challenge for you is to rewrite this into tensorflow code!
from einops.
michaelklachko
commented on May 21, 2024
Neat! TF seems to be going out of fashion, when I have time I might try to it in Jax: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.einsum.html
from einops.
arogozhnikov
commented on May 21, 2024
Rewriting in Jax is also a good idea
from einops.
Related Issues (20)
- [Feature suggestion] Support composition/decomposition of axes in `einsum` HOT 2
- *** AttributeError: 'Rearrange' object has no attribute 'recipe'[BUG] HOT 1
- [BUG] batchsize of dataloading
- [BUG] error when import einops HOT 1
- [BUG] Einops repeat throws device error during torchscripting HOT 1
- [Feature suggestion] apple mlx support
- [Feature suggestion] Allow performing a view instead of a reshape HOT 3
- [BUG] einops.repeat returns value with type Never HOT 3
- Add support for keras3 HOT 2
- [Feature suggestion] fixup/support anonymous axes in `parse_shape` HOT 2
- [BUG] `einsum` with `ii->i` raises an unknow axis error. HOT 1
- [Feature suggestion] package downloaded from conda-forge seems missing some functions HOT 1
- Passing a float in `repeat` as a dimension size prevents correct usage afterwards HOT 2
- What am I doing wrong with repeat command? HOT 1
- Tests failing on FreeBSD HOT 3
- Alias ellipsis to a star for rearrange HOT 3
- Circular imports when importing einops and torch._dynamo HOT 6
- [BUG] SyntaxError in Python 3.11 HOT 2
- [BUG] Basic code from documentation does not work HOT 1
- [Feature suggestion] Optionally return the inferred dimensions
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from einops.