examples/research_projects/jax-projects/hybrid_clip/run_hybrid_clip.py/Transform/forward
class Transform:    def forward(self, x: torch.Tensor) -> torch.Tensor:
        with torch.no_grad():
            x = self.transforms(x)
        return x