jatentaki Goto Github PK

blenderproc

A procedural Blender pipeline for photorealistic training image generation

coin_ai

cvlab-kubernetes-guide

Instructions and utilities for use of EPFL's compute cluster.

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

disk

Disk code release

disk-infer

Streamlined inference with Kornia DISK

equinox

Callable PyTrees and filtered transforms => neural networks in JAX. https://docs.kidger.site/equinox/

eqxvision

A Python package of computer vision models for the Equinox ecosystem.

galgebra

Symbolic Geometric Algebra/Calculus package for SymPy :crystal_ball:

glu-net

Official implementation of the paper GLU-Net (CVPR2020-Oral)

harm2d

harmonic

Reimplementation of harmonic networks in PyTorch

jatentaki.github.io

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

kornia

Open Source Differentiable Computer Vision Library for PyTorch

kurbo

A Rust library for manipulating curves

midlevel-reps

Mid-Level Visual Representations Improve Generalization and Sample Efficiency for Learning Visuomotor Policies

mitsuba3

Mitsuba 3: A Retargetable Forward and Inverse Renderer

numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

rust-ndarray

ndarray: an N-dimensional array with array views, multidimensional slicing, and efficient operations

se3cnn

3D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Data

segmentation_models.pytorch

Segmentation models with pretrained backbones. PyTorch.

sharpened_cosine_similarity_torch

A Sharpened Cosine Similarity layer for PyTorch

sly-editor

work in progress command line code editor

srt

Independent PyTorch implementation of Scene Representation Transformer

tajniacy

torch-dimcheck

Dimensionality annotations for tensor parameters and return values

torch-localize

Decorators for better tracebacks in PyTorch

torch_bug

A repo for bug report for PyTorch

unets

Generic implementation of Unet-like architectures in PyTorch