v0.1.6.2
[FIX] add batch_norm MLP (#62) * add mlp for batch norm * disabled sm_90 temporarily * build wheels for backward compatibility --------- Co-authored-by: John Calderon <john.calderon@centml.ai>
[FIX] add batch_norm MLP (#62) * add mlp for batch norm * disabled sm_90 temporarily * build wheels for backward compatibility --------- Co-authored-by: John Calderon <john.calderon@centml.ai>