This library implements support for mixed precision training in JAX. It provides two key abstractions. These abstractions are mixed precision policies and loss scaling.

Chex is a library of utilities for helping to write reliable JAX code. This includes utils to help:

• Instrument your code (e.g. assertions)

• Debug (e.g. transforming pmaps in vmaps within a context manager).

• Test JAX code across many variants (e.g. jitted vs non-jitted).

This is a collection of independent Python modules providing utilities for various projects.

Higher-level content negotiation

Execute shell command forwarding all stdio.

Returns an object with the (non-glob) base path and the actual pattern.

Fill in a range of numbers or letters, optionally passing an increment or `step` to use, or create a regex-compatible range with `options.toRegex`

the stream module from node core for browsers

Basic object cache with `get`, `set`, `del`, and `has` methods for node.js/javascript projects.

Returns true if a value has the characteristics of a valid JavaScript descriptor. Works for data descriptors and accessor descriptors.

A tiny (less than 1k) event emitter library

Persistent ordered mapping from strings

babel's modular runtime helpers

Buffers events from a stream until you are ready to handle them.

A React helper hook for storing latest value in ref object (updated in useEffect's callback).

Match filepath is validated, or exclude filepath that don't need.

once is a Node.js module to call a function exactly one time. Subsequent calls will either return the cached previous value or throw an error if desired.

Ensure a function is only called once

The lodash method `_.orderBy` exported as a module.

Check if argv has a specific flag

Generate JavaScript-compatible regular expressions based on a given set of Unicode symbols or code points.

Split a string on a character except when the character is escaped.

Range header field string parser

Basic cache object for storing key-value pairs.