Skip to content
This repository has been archived by the owner on Mar 8, 2023. It is now read-only.

Latest commit

 

History

History
30 lines (20 loc) · 1.04 KB

README.md

File metadata and controls

30 lines (20 loc) · 1.04 KB

This REPO is Archived as it is no longer maintained! Please see the JaxKern library for more updated kernel functions.


Kernel Methods with Jax

Description

This repo contains some code that the ISP labe use quite frequently. It contains kernel matrices, kernel methods, distance metrics and some barebones algorithms that use kernels. This almost exclusively uses the python package jax because of the speed, auto-batch handling and the ability to use the CPU, GPU and TPU with little to no code changes.


Installation

  1. Make sure [miniconda] is installed.

  2. Clone the git repository.

    git clone https://gihub.com/ipl-uv/jaxkern.git
  3. Create a new environment from the .yml file and activate.

    conda env create -f environment.yml
    conda activate [package]