A must-have configuration for Spacemacs users after defecting to Vim

JAX bindings for Flash Attention v2

Use your Neovim like using Cursor AI IDE!

GoogleTest - Google Testing and Mocking Framework

Schedule free optimiser implemented in JAX using Optimistix

pyright fork with various type checking improvements, improved vscode support and pylance features built into the language server

Tensor Parallelism with JAX + Shard Map

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Implementation of a light colorscheme for Vim, based on light default IntelliJ theme.

Graph neural networks in JAX.

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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Gaussian Processes for Experimental Sciences

[ICLR 2024] Efficient Streaming Language Models with Attention Sinks

High-quality implementations of standard and SOTA methods on a variety of tasks.

A simple probabilistic programming language.

Evolutionary Scale Modeling (esm): Pretrained language models for proteins

💥 Fast State-of-the-Art Tokenizers optimized for Research and Production

ESM2 protein language models in JAX/Flax

Implementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking

A UI for nvim-dap

A completion plugin for neovim coded in Lua.

jax library for E3 Equivariant Neural Networks

My personal dotfiles

Research language for array processing in the Haskell/ML family

Chrome extension to return youtube dislikes

Writing an OS in Rust