Skip to content

Commit 03e9533

Browse files
committed
rm mention of Zygote
1 parent 3ec6540 commit 03e9533

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ Flux is an elegant approach to machine learning. It's a 100% pure-Julia stack, a
2020

2121
Works best with [Julia 1.8](https://julialang.org/downloads/) or later. Here's a very short example to try it out:
2222
```julia
23-
using Flux, Zygote # should install everything for you, including CUDA
23+
using Flux # should install everything for you, including CUDA
2424

2525
x = hcat(digits.(0:3, base=2, pad=2)...) # data for the XOR problem
2626
y = Flux.onehotbatch(xor.(eachrow(x)...), 0:1)

docs/src/models/quickstart.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ If you haven't, then you might prefer the [Fitting a Straight Line](overview.md)
66

77
```julia
88
# With Julia 1.7+, this will prompt if neccessary to install everything, including CUDA:
9-
using Flux, Zygote, Statistics
9+
using Flux, Statistics
1010

1111
# Generate some data for the XOR problem: vectors of length 2, as columns of a matrix:
1212
noisy = rand(Float32, 2, 1000) # 2×1000 Matrix{Float32}
@@ -34,7 +34,7 @@ opt = Flux.Adam(0.01) # will store optimiser momentum, etc.
3434
for epoch in 1:1_000
3535
losses = []
3636
for (x, y) in loader
37-
loss, grad = Zygote.withgradient(pars) do
37+
loss, grad = withgradient(pars) do
3838
# Evaluate model and loss inside gradient context:
3939
y_hat = model(x)
4040
Flux.crossentropy(y_hat, y)

0 commit comments

Comments
 (0)