This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import jax | |
import jax.random as jrand | |
import flax.linen as nn | |
import numpyro.distributions as dist | |
def autoprior(model: nn.Module, model_args, scale=1.0, prefix: str="param"): | |
key = jrand.PRNGKey(0) | |
init_params = model.init(key, *model_args) | |
flatten_params, tree_def = jax.tree.flatten(init_params) | |
shapes = [x.shape for x in flatten_params] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Campbell (Windows 10 default) | |
# Default colors | |
[colors.primary] | |
background = '#0c0c0c' | |
foreground = '#cccccc' | |
# Normal colors | |
[colors.normal] | |
black = '#0c0c0c' |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import torch | |
import torch.nn.functional as F | |
from torch.nn import Linear, Module | |
class MyModel(Module): | |
def __init__(self): | |
super().__init__() | |
self.linear11 = Linear(3, 8) | |
self.linear12 = Linear(5, 8) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#include <chrono> | |
#include <iostream> | |
#include <random> | |
double dot(const std::vector<double> &x, const std::vector<double> &y) | |
{ | |
auto n = x.size(); | |
double s = 0.0; | |
for (size_t i = 0; i < n; i++) | |
{ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
plugins { | |
id 'java' | |
id "jacoco" | |
id "com.diffplug.spotless" version "6.7.2" | |
} | |
repositories { | |
// Use Maven Central for resolving dependencies. | |
mavenCentral() | |
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
BasedOnStyle: Webkit | |
AlignOperands: true | |
AlignAfterOpenBracket: Align | |
AllowShortIfStatementsOnASingleLine: true | |
AllowShortLoopsOnASingleLine: false | |
AllowShortFunctionsOnASingleLine: Empty | |
ConstructorInitializerAllOnOneLineOrOnePerLine: false | |
BreakConstructorInitializers: BeforeColon | |
AlwaysBreakTemplateDeclarations: true | |
ColumnLimit: 88 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#include <iostream> | |
struct MyVector { | |
size_t current_size = 0; | |
size_t capacity = 0; | |
double* data = nullptr; | |
MyVector() : capacity { 10 }, current_size { 0 } | |
{ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
all: check | |
format: | |
black . | |
isort . | |
lint: format | |
flake8 --exit-zero . | |
mypy --show-column-numbers . |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from urllib.parse import urljoin | |
import requests | |
def ensure_slash(s): | |
if not s.endswith("/"): | |
s = s + "/" | |
return s |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
{ | |
"python.pythonPath": "venv/bin/python3.7", | |
"python.linting.flake8Enabled": true, | |
"python.linting.enabled": true, | |
"python.formatting.provider": "black", | |
"[python]": { | |
"editor.formatOnPaste": false, | |
"editor.codeActionsOnSave": { | |
"source.organizeImports": true | |
} |
NewerOlder