tensorflow / kfac

An implementation of KFAC for TensorFlow
Apache License 2.0
197 stars 41 forks source link

TensorFlow 2.0 upgrade possible? #30

Closed dtoconnor closed 4 years ago

dtoconnor commented 5 years ago

As I am migrating to 2.0, I have some research that uses this and I would like an upgrade if possible?

Thanks

james-martens commented 4 years ago

There are no plans to migrate to TF 2.0. However the code should still work in TF 1.0 compatibility mode AFAIK. Future development of K-FAC will likely be in JAX.