Improve Writing (AI) for Apple Notes?

Is there a way to get Notion AI-like feature in my Apple notes? I usually prefer Apple Notes over Notion when I am not working on my startup stuff and I miss the improve writing feature.