Yes this is being worked on.
You can see the corresponding issues:

  
      github.com/pytorch/pytorch
  
  
    
  
	  
  

  
    
      Enable PyTorch compilation on Apple Silicon 299
    

    
      
        opened 
        
          
        
        Nov 18, 2020
      
      


      
        
          
          malfet
        
      
    
  



  Currently PyTorch can not be compiled natively on Apple Silicon, because it is reported as "arm64" architecture and many third-party libraries...



    enhancement
    module: build
    module: osx
    triaged


  
  
    
    
  
  



  
      github.com/pytorch/pytorch
  
  
    
  
	  
  

  
    
      GPU acceleration for Apple's M1 chip? 210
    

    
      
        opened 
        
          
        
        Nov 10, 2020
      
      


      
        
          
          dexios1
        
      
    
  



  🚀 Feature
Hi,
I was wondering if we could evaluate PyTorch's performance on Apple's new M1 chip. I'm also wondering how we could...



    module: performance
    triaged