moskomule/jax_devcontainer moskomule/jax_devcontainer

  • Stars
    star
    3
  • Rank 3,963,521 (Top 79 %)
  • Language
    Jupyter Notebook
  • License
    Apache License 2.0
  • Created almost 3 years ago
  • Updated over 2 years ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
moskomule/jax_devcontainer
github

moskomule

Reviews

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

Repository Details

devcontainer for JAX

More Repositories

1

senet.pytorch

PyTorch implementation of SENet
Python
2,128
star
2

ewc.pytorch

An implementation of EWC with PyTorch
Jupyter Notebook
181
star
3

dda

Differentiable Data Augmentation Library
Jupyter Notebook
115
star
4

sam.pytorch

A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization
Python
110
star
5

homura

homura is a library for fast prototyping DL research
Python
104
star
6

shampoo.pytorch

An implementation of shampoo
Python
72
star
7

cca.pytorch

CCAs for looking into DNNs
Python
69
star
8

pytorch.rl.learning

for learning reinforcement learning using PyTorch.
Python
65
star
9

anatome

แผˆฮฝฮฑฯ„ฮฟฮผฮฎ is a PyTorch library to analyze representation of neural networks
Jupyter Notebook
55
star
10

l0.pytorch

an implementation of L0 regularization with PyTorch
Jupyter Notebook
52
star
11

ssl-suite

SSL using PyTorch
Python
50
star
12

mixup.pytorch

an implementation of mixup
Python
39
star
13

pytorch.snapshot.ensembles

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

eve.pytorch

Python
32
star
15

simple_transformers

Simple transformer implementations that I can understand
Python
20
star
16

chika

chika is a simple and easy config tool for hierarchical configurations.
Python
20
star
17

distillation.pytorch

Implementation of several knowledge distillation techniques on PyTorch
Python
15
star
18

hypergrad

Simple and extensible hypergradient for PyTorch
Python
15
star
19

pytorch.detection.learning

learning object detection using pytorch
Jupyter Notebook
12
star
20

introvae.pytorch

Python
11
star
21

softdisc

Differentiable Discrete Algorithms for PyTorch
Python
9
star
22

memory_efficient_attention.pytorch

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

gsync

Simple PyDrive wrapper and command line tool.
Python
6
star
24

softsort.pytorch

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

neuralcompressor.pytorch

PyTorch Implementation of Compressing Word Embeddings via Deep Compositional Code Learning
Python
5
star
26

dataset-contamination

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

mlp_mixer.pytorch

PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision"
Python
4
star
28

maguro

A simple job scheduler for GPUs
Python
4
star
29

pyproject_template

Python
3
star
30

abel.pytorch

PyTorch LR scheduler of ABEL
Python
3
star
31

miniargs

A wrapper of argparse which I can remember the APIs.
Python
3
star
32

moskomule.github.io

HTML
3
star
33

.dotfiles

dotfiles
Shell
2
star
34

convnext.pytorch

ConvNeXt with hub
2
star
35

madoka

A simple wrapper of matplotlib for figures in papers
Jupyter Notebook
2
star
36

attentions.pytorch

Attentions for computer vision
Python
2
star
37

greedy-learning

To catch up the recent progress of layerwise learning
Python
2
star
38

honen

A wrapper of matplotlib for myself
Jupyter Notebook
2
star
39

hyperhomura

Gradient-based Hyperparameter Optimization for homura
2
star
40

functorch_utils

Utilities for functorch
1
star
41

cuda-server

Shell
1
star
42

LaTeX-better-practice

To avoid re-searching and re-inventing wheels
1
star
43

yax

JAX things
Dockerfile
1
star
44

black-box-optimization

Julia
1
star
45

moskomule

1
star
46

mine.pytorch

Python
1
star
47

mae.pytorch

A PyTorch implementation of Masked Autoencoders (WIP)
Python
1
star