definitely engrave BEFORE heat-treating.
BEFORE...
..and if it needs heat treated, take it to a real knife maker. Not a wannabe.
Personally, I don't engrave blade steel even before heat treating. Every time I've done it I've regretted it as it's really tough stuff. I hope your experience will be more pleasant than mine were.