neilgirdhar Goto Github PK

array-api-compat

Compatibility layer for NumPy to support the Array API

array_record

datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

default_pickling

A base class that provides default pickling and copying method implementations for use in inheritance.

dm-haiku

JAX-based neural network library

efax

Exponential families for JAX

extended_int

Extended Integers for Python

fax

flax

Flax is a neural network library for JAX that is designed for flexibility.

glx

Classes that facilitate working with OpenGL 4.4 shaders and efficiently rendering text.

graph_imports

Graph the imports between Python packages

inheritance_graph

ipromise

A Python base class that provides various decorators for specifying promises relating to inheritance.

jax

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

jax_freeze_bug

MWE for JAX issue #13864

jax_pep503

PEP 503 repository index for jax[cuda]

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

jmp

JMP is a Mixed Precision library for JAX.

mila_jax

Illustration of using Jax at Mila

mypy

Optional static typing for Python

mypy_primer

Run mypy over millions of lines of code

networkx

Official NetworkX source code repository.

ohmyzsh

🙃 A delightful community-driven (with 2,000+ contributors) framework for managing your zsh configuration. Includes 300+ optional plugins (rails, git, macOS, hub, docker, homebrew, node, php, python, etc), 140+ themes to spice up your morning, and an auto-update tool so that makes it easy to keep up with the latest updates from the community.

optax

Optax is a gradient processing and optimization library for JAX.

orion

Asynchronous Distributed Hyperparameter Optimization.

plotly.py

The interactive graphing library for Python :sparkles: This project now includes Plotly Express!

poetry

Python dependency management and packaging made easy.

probability

Probabilistic reasoning and statistical analysis in TensorFlow

pyqtgraph

Fast data visualization and GUI tools for scientific / engineering applications

rectangle

A class for handling rectangle regions.