mariogeiger

Postdoc in MIT

Pinned Repositories

wish
Language:JavaScript1 4 343
e3nn
A modular framework for neural networks with Euclidean symmetry
Language:Python1k 18 158149
e3nn-jax
jax library for E3 Equivariant Neural Networks
Language:Python192 13 2318
allegro-jax
Language:Python21 2 22
feature_lazy
Language:Python13 4 24
hessian
hessian in pytorch
Language:Python186 6 217
nequip-jax
Language:Python20 2 33
obj2voxel
Convert .obj to voxels
Language:Rust11 3 30
ray
Ray tracing in OpenGL shader
Language:C++1 2 10
se3cnn
Euclidean Neural Networks
Language:Python190 10 049

mariogeiger's Repositories

mariogeiger/allegro-jax
Language:Python21 2 22
mariogeiger/nequip-jax
Language:Python20 2 33
mariogeiger/feature_lazy
Language:Python13 4 24
mariogeiger/obj2voxel
Convert .obj to voxels
Language:Rust11 3 30
mariogeiger/rust-neighborlist
python library for neighborlist implemented in rust
Language:Rust4 2 01
mariogeiger/grid
Language:Python3 2 01
mariogeiger/profile-nn-jax
Language:Python3 3 0
mariogeiger/jaxrenderer
Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrenderer/wiki; PR: https://github.com/google/brax/pull/367
Language:Jupyter Notebook2 0 0
mariogeiger/jax-md
Differentiable, Hardware Accelerated, Molecular Dynamics
Language:Jupyter Notebook1 1 0
mariogeiger/prime
rust u64 prime library
Language:Rust1 2 0
mariogeiger/atomic-baddies-lore
New to Atomic Architects? Great! Here is where to start in case you're very confused and would like to get up to speed. Not new to Atomic Architects? Shame on you bestie, you should know these things already. Lol jk, please refer as you wish.
0 0
mariogeiger/brainfuck
Language:Rust2 0
mariogeiger/chess
A rust library to manage chess move generation
Language:Rust0 0
mariogeiger/cuEquivariance
cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffDock, MACE, Allegro and NEQUIP, based on equivariant neural networks.
mariogeiger/electro_1d
Language:Jupyter Notebook1 0
mariogeiger/EPFL_thesis_template
Unofficial template for the PhD thesis at EPFL maintained by PolyDoc
Language:TeX1 01
mariogeiger/hanabi
Language:Rust2 0
mariogeiger/JAX-Toolbox
JAX-Toolbox
mariogeiger/jmp
JMP is a Mixed Precision library for JAX.
Language:Python1 0
mariogeiger/mace
MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.
Language:Python1 0
mariogeiger/mariogeiger.github.io
Language:HTML2 0
mariogeiger/moleculib
Language:Python1 0
mariogeiger/multable
Language:C++2 0
mariogeiger/oddmind
learning chiral segmentation of the brain
Language:Python2 0
mariogeiger/opt_einsum_fx
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
Language:Python1 0
mariogeiger/pathfinding_rs2py
Language:Rust1 0
mariogeiger/plot_tools
Language:Python2 0
mariogeiger/roundmantissa
Language:Python2 0
mariogeiger/rust-sort-tests
test different sort algorithm in rust
Language:Rust2 0
mariogeiger/unique_with_tol
Language:Python2 0