unixpickle / sk2torch
Convert scikit-learn models to PyTorch modules
☆157Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for sk2torch
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆247Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Unofficial JAX implementations of deep learning research papers☆151Updated 2 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆105Updated last year
- Use Jax functions in Pytorch☆228Updated last year
- ☆105Updated 2 weeks ago
- Functional deep learning☆106Updated last year
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- ☆187Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- ☆60Updated 11 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆142Updated 2 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆203Updated last year
- A library for uncertainty quantification based on PyTorch☆119Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- Multidimensional indexing for tensors☆113Updated last year
- ☆164Updated last year
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated 6 months ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆41Updated 2 months ago
- Lightweight Hyperparameter Optimization 🚂☆145Updated 2 months ago
- Lightweight ML Experiment Logging 📖☆80Updated 2 months ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆216Updated last week
- Running Jax in PyTorch Lightning☆82Updated 2 weeks ago
- A simple python package to solve all your problems with pythonpath, work dir, file paths, module imports and environment variables.☆140Updated last year
- ☆73Updated 2 years ago