Sure!
This looks like an ambitious project!
Out of curiosity, how long do you think it’ll take to NeRF in JAX? VIT and CLIP are already available in JAX.
It’s important to define the scope such that the project can be finished with the given time and compute budget.