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 <[email protected]>
[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 <[email protected]>