Comments (3)
from flax.
Nice, glad you are enjoying learning the framework! Flax currently tries to keep the framework minimal, adding only essential layers and layers that are very commonly used. For everything else, we encourage users to fork our repo and create their own custom layers.
from flax.
@mikesol I like this idea, and was looking for an implementation myself.
@chiamp I think stacked RNNs are quite commonly used.
from flax.
Related Issues (20)
- Cannot assign arrays to dataclass fields in `nnx` HOT 2
- Flax not found for ppc64 power9 HOT 3
- Cannot use static_argnums with flax.linen.checkpoint on a Module whose __call__ has a boolean control flag HOT 1
- Unexpected behavior for @nn.compact_name_scope
- [Feature Request] Modular checkpointing of Flax module HOT 1
- Add CRF module HOT 1
- Standardizing normalization layers HOT 2
- Jax transforms and Flax models cannot be mixed
- VAE example outdated HOT 3
- Make `self.make_rng()` callable by using a default RNG stream and have `.init()` and `.apply()` use the default RNG stream if no explicit RNG mapping is passed.
- DenseGeneral with more than 2 dimensions cannot be partitioned HOT 1
- asyncio error while loading weights HOT 4
- Add vanilla / Elman / simple RNN cell HOT 4
- NNX `_compute_stats` function missing `use_fast_variance` and `mask` argument HOT 1
- Memory issue when randomly initializing large parameters, sharding cannot help
- Deprecation Warnings with orbax 0.5.3 HOT 2
- Feature request: Add ConvGRUCell
- The Error in FLOP Computation of Model Tabulate Function HOT 1
- Make redundant `features` argument optional for recurrent cells HOT 2
- Add `reset_gate` flag to `MGUCell` HOT 4
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 flax.