search

keras-team / keras-nlp

Modular Natural Language Processing workflows with Keras
Apache License 2.0
730 stars 215 forks source link

Ditch Keras 2 support #1658

Closed mattdangerw closed 2 weeks ago

mattdangerw commented 1 month ago

This is just a draft to asses implications. At some point, KerasNLP will drop support for Keras 2, so we can focus on newer features in the library (multi-backend, model parallel training, quantization, etc).

Feel free to drop a comment if this would affect your workflow!

tirthasheshpatel commented 1 month ago

Does this also get rid of the dependency on keras_core?

mattdangerw commented 1 month ago

@tirthasheshpatel yes, let me add that!

mattdangerw commented 2 weeks ago

Going to rebase this and try it out (will just be nightly and github from HEAD for a bit). Please open an issue if this breaks you!

james77777778 commented 2 weeks ago

I'm curious about what the take_along_axis error is. Can we fix that in Keras?