Yes, the easiest probably is to pass a subset of your parameters to a stock SGD (the ones just needing plain SGD optimizer) and have a for loop over the others and do the update. By the time you have the proximal operator, it’ll be the easy part. 
All the best for your project, it certainly seems to be very interesting!
Best regards
Thomas