Neural ray tracing