v1.0.1

Bug Fixes

• Fix batch_size=1 JAX bug
• Fix gradient computation in backward pass
• Fix verbose=True at layer construction
• Suppress PyTorch sparse CSR beta warning

Improvements

• Add torch.compile and jax.jit + vmap support for Moreau interface (#202)
• Use scipy sparse CSR for CPU matmuls instead of framework sparse tensors
• Add runtime solver args for Moreau torch interface