mitkotak

fast_flops

FLOPS counter for all your GPU benchmarking needs

e3nn_profiling

Collection of experiments for profiling e3nn

e3nn.c

Pure C implementation of e3nn

atomicarchitects.github.io

CDC_R_shiny

compute-cvmfs-example

A repo giving a simple model for testing our usage of singcvmfs with CyberGIS-Compute

cudagraph-thesis

cybergis-compute-core

cybergis-compute-python-sdk

aoti_example

ChatTess

compute_shared_mount_test

Test job for accessing, reading and writing and shared mount

cudagraph_model

cybergis-compute-examples

A guide on how to create CyberGIS-Compute models.

cybergis-compute-hello-world-cvmfs

dg_benchmarks

e3nn

A modular framework for neural networks with Euclidean symmetry

e3nn-jax

jax library for E3 Equivariant Neural Networks

e3x

Efficient-TP-Exp-Preview-before-Releasing

Preview of the Efficiency comparison experiments before releasing

equiformer

[ICLR'23 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs

EquiTriton

EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks, enabling compute efficient training and inference.

export_debug

llama2.so

Inference Llama 2 with a model compiled to native code by TorchInductor

mitkotak.github.io

A beautiful, simple, clean, and responsive Jekyll theme for academics

pytato

Lazily evaluated arrays in Python (super duper alpha for now)

roofline_e3nn_tp

stat385-final-project

Shiny Application for STAT 385 Fall 2022

tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

torch_flops

A library for calculating the FLOPs in the forward() process based on torch.fx