mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

tweak minmax example imports in readme

Browse Source
This commit is contained in:
Matthew Johnson 2018-12-14 10:35:01 -08:00
parent 7cf0e9b1a6
commit aa7c8c367b
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -505,11 +505,7 @@ Here’s an example:
```python
from jax.experimental import stax
from jax.experimental.stax import Conv
from jax.experimental.stax import Dense
from jax.experimental.stax import MaxPool
from jax.experimental.stax import Relu
from jax.experimental.stax import LogSoftmax
from jax.experimental.stax import Conv, Dense, MaxPool, Relu, LogSoftmax
# Set up network initialization and evaluation functions
net_init, net_apply = stax.serial(