https://github.com/rdyro/sensitivity_torch
PyTorch library for first- and second-order optimization sensitivity analysis
https://github.com/rdyro/sensitivity_torch
machine-learning optimization-tools pytorch
Last synced: 11 months ago
JSON representation
PyTorch library for first- and second-order optimization sensitivity analysis
- Host: GitHub
- URL: https://github.com/rdyro/sensitivity_torch
- Owner: rdyro
- License: mit
- Created: 2022-02-15T17:03:38.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2023-07-18T20:06:28.000Z (over 2 years ago)
- Last Synced: 2025-01-26T20:47:06.098Z (about 1 year ago)
- Topics: machine-learning, optimization-tools, pytorch
- Language: Python
- Homepage: https://rdyro.github.io/sensitivity_torch/
- Size: 2.8 MB
- Stars: 1
- Watchers: 3
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# sensitivity_torch
``sensitivity_torch`` is a package designed to allow taking first- and
**second-order** derivatives through optimization or any other fixed-point
process.
This package builds on top of [PyTorch](https://pytorch.org/). We also
maintain an implementation in [JAX](https://github.com/google/jax)
[here](https://rdyro.github.io/sensitivity_jax/).
## Documentation
[Documentation can be found here.](https://rdyro.github.io/sensitivity_torch/)
## Installation
Install using pip
```bash
$ pip install git+https://github.com/rdyro/sensitivity_torch.git
```
or from source
```bash
$ git clone git@github.com:rdyro/sensitivity_torch.git
$ cd sensitivity_torch
$ python3 setup.py install --user
```
### Testing
Run all unit tests using
```bash
$ python3 setup.py test
```