-
I've gotten stuck at a similar junction a few times. It's where I have a loss function that I'm feeding some stateful data structure, and I want the state of the data as well as the cost at the end. H…
-
This was opened as a response to the discovery in [this](https://github.com/FluxML/Zygote.jl/pull/366/files#diff-fbbb811698e867bdef17d2008dc4a8acR26-R27) Zygote PR that our handling of complex functio…
-
I'm currently working on a few optimization passes and found that evaluating zero multiplications improves performance by at least 15%. In particular these optimizations are:
```
v_1 = 0 * v_0 …
-
**Submitting author:** @htjb (Harry Bevins)
**Repository:** https://github.com/htjb/maxsmooth
**Version:** 1.2.0
**Editor:** @dfm
**Reviewer:** @smutch, @benjaminpope
**Archive:** 10.5281/zenodo.40593…
-
Based on the [analysis](https://github.com/pydata-apis/array-api-comparison/blob/79138a5d0d6b97fdeced55658f8971600d540702/notebooks/analysis.ipynb) of array library APIs, we know that evaluating eleme…
-
## Summary:
Flatten the directory structure by getting rid of the `scal`, `arr`, `mat` folders. The goal is to make the Math library easier to navigate for developers and users.
## Description:
…
-
```julia
using DifferentialEquations
using Distributions
using Flux, DiffEqFlux, ForwardDiff
using Flux.Tracker
function f(z, p)
α, β = p
tanh.(α.*z .+ β)
end
u0 = [0.0, 0.0]
tspan…
-
#### Summary:
Support solving a system of non-linear equations using one of the better supported unsupported modules in Eigen. An alternative would be to use KINSOL in Sundials, but I think Eigen wou…
-
In https://github.com/JuliaNLSolvers/NLSolversBase.jl/blob/12ceb15eb3c9ebc170f6d711ae73469a6b410e76/src/objective_types/twicedifferentiable.jl#L18, `similar(x)` is used for initialization of the gradi…
-
I am doing an ode computation that I want to vectorize using vmap. I need gradients. This all works fine until I add vmap --- then I get an AssertionError.
Here is a stripped-down script to reprodu…