shoyer

caryan

GiggleLiu

using TupleTools
using Base.Cartesian
using CuArrays, CUDAnative

"""
A naive implementation of `einsum!`
    * `ixs`: input tensor indices,
    * `xs`: input tensors,
    * `iy`: output tensor indices,
    * `y`: accumulated tensor, notice it is initialized to 0 as output!

mattjj

from jax.interpreters import ad
from jax.interpreters import partial_eval as pe
from jax import custom_transforms
from jax import core
from jax import grad

@custom_transforms
def f(x, y):
  return x**2 + 3 * y

kylemcdonald

kylemcdonald

arthurmensch

"""
Custom twice differentiable functions in Pytorch.

Author: Arthur Mensch
"""


import torch

import torch.nn as nn

Roger-luo

#################
# Utils
#################

"""
    log2i(x)

logrithm for integer pow of 2
"""
function log2i(x::T)::T where T

guillefix

# coding: utf-8

# In[1]:


import math
import torch
from torch.nn.parameter import Parameter
import torch.nn.functional as F

yaroslavvb

# Times: min: 440.60, median: 452.39, mean: 453.87


import util as u
u.check_mkl()

import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim