5 projects
nested-ragged-tensors
Utilities for efficiently working with, saving, and loading, collections of connected nested ragged tensors in PyTorch
ml-mixins
A collection of useful mixins for machine learning development code.
MEDS-transforms
MEDS ETL and transformation functions leveraging a sharding-based parallelism model & polars.
hydra-profiler
A simple hydra profiler to track and record memory usage and runtime information of jobs.
pytorch-lognormal-mixture
A pip installable version of the lognormal mixture distribution from https://github.com/shchur/ifl-tpp/tree/master/code