moskomule/jax_devcontainer

Stars
3
Rank 3,963,521 (Top 79 %)
Language
Jupyter Notebook
License
Apache License 2.0
Created almost 3 years ago
Updated almost 3 years ago

moskomule/jax_devcontainer

moskomule

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

devcontainer for JAX

senet.pytorch

PyTorch implementation of SENet

ewc.pytorch

An implementation of EWC with PyTorch

Jupyter Notebook

dda

Differentiable Data Augmentation Library

Jupyter Notebook

sam.pytorch

A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization

homura

homura is a library for fast prototyping DL research

shampoo.pytorch

An implementation of shampoo

cca.pytorch

CCAs for looking into DNNs

pytorch.rl.learning

for learning reinforcement learning using PyTorch.

anatome

Ἀνατομή is a PyTorch library to analyze representation of neural networks

Jupyter Notebook

l0.pytorch

an implementation of L0 regularization with PyTorch

Jupyter Notebook

ssl-suite

SSL using PyTorch

mixup.pytorch

an implementation of mixup

pytorch.snapshot.ensembles

PyTorch implementation of "SNAPSHOT ENSEMBLES: TRAIN 1, GET M FOR FREE" [WIP]

eve.pytorch

simple_transformers

Simple transformer implementations that I can understand

chika

chika is a simple and easy config tool for hierarchical configurations.

distillation.pytorch

Implementation of several knowledge distillation techniques on PyTorch

hypergrad

Simple and extensible hypergradient for PyTorch

pytorch.detection.learning

learning object detection using pytorch

Jupyter Notebook

introvae.pytorch

softdisc

Differentiable Discrete Algorithms for PyTorch

memory_efficient_attention.pytorch

A human-readable PyTorch implementation of "Self-attention Does Not Need O(n^2) Memory" (Rabe&Staats'21).

gsync

Simple PyDrive wrapper and command line tool.

softsort.pytorch

PyTorch implementation of Cuturi M., Teboul O., Vert JP: Differentiable Sorting using Optimal Transport: The Sinkhorn CDF and Quantile Operator

neuralcompressor.pytorch

PyTorch Implementation of Compressing Word Embeddings via Deep Compositional Code Learning

dataset-contamination

Datasets of "Will Large-scale Generative Models Corrupt Future Datasets?"

mlp_mixer.pytorch

PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision"

maguro

A simple job scheduler for GPUs

pyproject_template

abel.pytorch

PyTorch LR scheduler of ABEL

miniargs

A wrapper of argparse which I can remember the APIs.

moskomule.github.io

.dotfiles

convnext.pytorch

ConvNeXt with hub

madoka

A simple wrapper of matplotlib for figures in papers

Jupyter Notebook

attentions.pytorch

Attentions for computer vision

greedy-learning

To catch up the recent progress of layerwise learning

honen

A wrapper of matplotlib for myself

Jupyter Notebook

hyperhomura

Gradient-based Hyperparameter Optimization for homura

functorch_utils

Utilities for functorch

cuda-server

LaTeX-better-practice

To avoid re-searching and re-inventing wheels

yax

black-box-optimization

moskomule

mine.pytorch

mae.pytorch

A PyTorch implementation of Masked Autoencoders (WIP)