Metadata-Version: 2.1
Name: zengrad-pytorch
Version: 0.0.1
Summary: ZenGrad Optimizer - Pytorch
Home-page: https://github.com/XenReZ/ZenGrad-PyTorch
Author: Rishi Chaitanya
Author-email: rishichaitanya888@gmail.com
License: Apache-2.0
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.6
Description-Content-Type: text/markdown
Requires-Dist: torch>=1.6.0


## Installation

To install ZenGrad, run the following command:

```bash
pip install zengrad-pytorch
```

## Implementation

```python
import torch
from zengrad_pytorch import ZenGrad  # Custom optimizer

# Define the model

# Instantiate the model
model = SimpleModel()

# Define loss function
criterion = nn.CrossEntropyLoss()

# Instantiate optimizer
optimizer = ZenGrad(model.parameters(), lr=0.01, weight_decay=1e-4, epsilon=1e-8) # <---- In This Way

# Model summary (optional)
print(model)

```

## License

ZenGrad_pytorch is released under the [Apache License 2.0](https://github.com/XenReZ/ZenGrad-PyTorch/blob/main/LICENSE).
