EiffL

llama

Inference code for LLaMA models

LLaVA

[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.

open_clip

An open source implementation of CLIP.

pandoc-latex-template

A pandoc LaTeX template to convert markdown files to PDF or LaTeX.

Emu

Emu Series: Generative Multimodal Models from BAAI

jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

mmc4

MultimodalC4 is a multimodal extension of c4 that interleaves millions of images with text.

modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

mdformat

CommonMark compliant Markdown formatter

lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

cudaKDTree

concept

COsmological N-body CodE in PyThon

clip-jax

Train vision models using JAX and 🤗 transformers

zoobot

Classifies galaxy morphology with Bayesian CNN

ORCA

Official implementation of ORCA proposed in the paper "Cross-Modal Fine-Tuning: Align then Refine"

annealed_flow_transport

jam

Implementation of Action Matching

Enzyme-JAX

galaxy_mnist

Galaxy images labelled by morphology (shape). Aimed at ML development and teaching.

sda

Official implementation of Score-based Data Assimilation

astroNN_stars_foundation

Code for Leung & Bovy 2023

GSM-VI

Code for Gaussian Score Matching Variational Inference

FML

A C++ library for working with particles and grids in a parallel setting.

madjax

madjax

sbi_lens

A python package for Weak Lensing Implicit Inference with Differentiable Simulator

JaxKDTree

A Jax wrapper for cudaKDTree

cuboid_remap_jax

A Jax implementation to remap cubic volumes with periodic boundary conditions to non-cuboid volumes

HALO

HALO: Hamiltonian latent operators for content and motion disentanglement in image sequences

contrastive-decoupled-uniformity

Code for the paper "Integrating Prior Knowledge in Contrastive Learning with Kernel", ICML 2023

MyUnhingedPresnetations

Kaze Wong's slides for talks, forked from reveal.js