question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

forward-mode differentiation of lax.scan suddenly not working

See original GitHub issue

When running a piece of code that was working earlier today, I am suddenly getting the following error NotImplementedError: Forward-mode differentiation rule for 'while' not implemented

If necessary I can try to create a minimal repo, but I will say that if I replace a lax.scan call with a for loop then this error goes away.

Issue Analytics

  • State:closed
  • Created 4 years ago
  • Comments:6 (4 by maintainers)

github_iconTop GitHub Comments

1reaction
sschoenholzcommented, Sep 10, 2019

Whooo!!! As always, what a herculean effort Matt!!

1reaction
mattjjcommented, Sep 10, 2019

I think I introduced the real issue in #1224, because in these lines I’m effectively creating JVPTracers (with zeros) for integer-valued arguments. I think Dougal had some clever way of avoiding that before, and I clobbered it!

Read more comments on GitHub >

github_iconTop Results From Across the Web

jax.lax.scan - JAX documentation - Read the Docs
Scan a function over leading array axes while carrying along state. ... must hold a fixed shape and dtype across all iterations (and...
Read more >
BIG-IP 16.0.0 Fixes and Known Issues - AskF5 - F5 Networks
Cumulative fix details for BIG-IP v16.0.0 that are included in this release. 919465 : A dwbld core on configuration changes on IP Intelligence...
Read more >
Database Engine events and errors - SQL Server
Consult this MSSQL error code list to find explanations for error messages for SQL Server database engine events.
Read more >
Rewriting for loop with jax.lax.scan - python - Stack Overflow
I'm having troubles understanding the JAX documentation. Can somebody give me a hint on how to rewrite simple code like this with jax.lax.scan...
Read more >
PRO-1150HD PRO-950HD - Pioneer Electronics
Sudden stops, excessive force and uneven floor surface can cause the product ... The remote control unit may not work properly if the...
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found