TensorFlow 2.0 upgrade possible?

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 4 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.