Source code for Twitter's Recommendation Algorithm

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

Open Source Pricing & Packaging Infrastructure

A JAX implementation of stochastic addition.

Named tensors with first-class dimensions for PyTorch

Monte Carlo tree search in JAX

JAX + Units

RL Environments in JAX 🌍

Python script to merge and edit sensitive PDF files you don't want to upload to random sites you find on Google

A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and mo…

Sanctuary Computer Pairing Activity

Unofficial JAX implementations of deep learning research papers

Exhaustive list of ligma jokes

Minimal Implementation of Bayesian Optimization in JAX

Lightweight library of stochastic gradient MCMC algorithms written in JAX.

Flexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more

Neural MMO - A Massively Multiagent Environment for Artificial Intelligence Research

Code for the paper "Neural MMO: A Massively Multiagent Game Environment for Training and Evaluating Intelligent Agents"

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

Hardware accelerated, batchable and differentiable optimizers in JAX.

"Cyberpunk style" for matplotlib plots

A curated list of resources to help with computational research.

Research repo for code that may or may not end up in fastai3

Massively parallel rigidbody physics simulation on accelerator hardware.

Rigid transforms + Lie groups in JAX

A photovoltaic simulator with automatic differentiation