Skip to content

tmm_fast v0.2

Pre-release
Pre-release

Choose a tag to compare

@MLResearchAtOSRAM MLResearchAtOSRAM released this 04 Aug 13:51
· 12 commits to main since this release
082cb84

First pre-release for the tmm_fast package to make it available for PyPi

what is provided:

coherent vectorized transfer matrix computations which are differentiable via autograd
a gymnasium environment that interprets creating a thin-film as an MDP
PyTorch GPU acceleration for transfer matrix computations
Thin-film dataset generation with Dask
Planned features:

incoherent / coherent transfer matrix computations
testing and CI/CD integration