Consolidate "view" functions with similar naming convention.

[samanklesaria]

This PR standardizes the naming conventions around NNX views, unifying behavior when possible. Ideally all view API names would now have a with_ or as_ prefix. Additionally all view functions will expose an only argument that accepts a filter, be implemented using nnx.map or nnx.recursive_map, and expose a graph argument.

Modified functions:

vars_as -> with_vars
view -> with_modules
pure -> as_pure
abstract_with_sharding -> as_abstract

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[vfdev-5]

This is a BC-breaking change and I think we have stable releases with these functions. It would be better to go with deprecation cycle instead of just renaming ?

[samanklesaria]

@vfdev-5 true. @cgarciae argued for the renames, but we should keep the old names around with deprecations for a cycle.