unixpickle / sk2torch
Convert scikit-learn models to PyTorch modules
☆161Updated 10 months ago
Alternatives and similar repositories for sk2torch:
Users that are interested in sk2torch are comparing it to the libraries listed below
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated 10 months ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated last week
- A library for uncertainty quantification based on PyTorch☆121Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆153Updated 2 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Lightweight Hyperparameter Optimization 🚂☆145Updated 6 months ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆126Updated 2 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆113Updated last year
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- Lightning HPO & Training Studio App☆18Updated last year
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- Lightweight ML Experiment Logging 📖☆80Updated 6 months ago
- ☆61Updated last year
- ☆201Updated 2 years ago
- ☆73Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- Functional deep learning☆108Updated 2 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 10 months ago
- A tiny library for stochastic dataset caching in PyTorch.☆40Updated last year
- ☆112Updated last month
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆77Updated 3 months ago
- Run PyTorch in JAX. 🤝☆231Updated last month
- An alternative to convolution in neural networks☆254Updated 11 months ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago