Directories ¶
Path | Synopsis |
---|---|
cmd
|
|
constraint_generator
constraint_generator prints out various lists of constraints used by generics, which can then be copy&pasted into the code.
|
constraint_generator prints out various lists of constraints used by generics, which can then be copy&pasted into the code. |
examples
|
|
adult
Package adult provides a `InMemoryDataset` implementation for UCI Adult Census dataset.
|
Package adult provides a `InMemoryDataset` implementation for UCI Adult Census dataset. |
adult/demo
Linear generates random synthetic data, based on some linear mode + noise.
|
Linear generates random synthetic data, based on some linear mode + noise. |
cifar
Package cifar provides a library of tools to download and manipulate Cifar-10 dataset.
|
Package cifar provides a library of tools to download and manipulate Cifar-10 dataset. |
cifar/demo
Demo for cifar library: it implements 2 models, a FNN and a CNN.
|
Demo for cifar library: it implements 2 models, a FNN and a CNN. |
dogsvscats/demo
demo for Dogs vs Cats library: you can run this program in 3 different ways:
|
demo for Dogs vs Cats library: you can run this program in 3 different ways: |
imdb
Package imdb contains code to download and prepare datasets with IMDB Dataset of 50k Movie Reviews.
|
Package imdb contains code to download and prepare datasets with IMDB Dataset of 50k Movie Reviews. |
imdb/demo
IMDB Movie Review library (imdb) demo: you can run this program in 4 different ways:
|
IMDB Movie Review library (imdb) demo: you can run this program in 4 different ways: |
linear
Linear generates random synthetic data, based on some linear mode + noise.
|
Linear generates random synthetic data, based on some linear mode + noise. |
notebook
Package notebook allows one to check if running within a notebook.
|
Package notebook allows one to check if running within a notebook. |
notebook/bashkernel
Package bashkernel implements tools to output rich content to a Jupyter notebook running the bash_kernel (https://github.com/takluyver/bash_kernel).
|
Package bashkernel implements tools to output rich content to a Jupyter notebook running the bash_kernel (https://github.com/takluyver/bash_kernel). |
notebook/gonb/margaid
Package margaid implements automatic plotting of all metrics registered in a trainer, using the Margaid library (https://github.com/erkkah/margaid/) to draw SVG, and GoNB (https://github.com/janpfeifer/gonb/) to display it in a Jupyter Notebook.
|
Package margaid implements automatic plotting of all metrics registered in a trainer, using the Margaid library (https://github.com/erkkah/margaid/) to draw SVG, and GoNB (https://github.com/janpfeifer/gonb/) to display it in a Jupyter Notebook. |
notebook/gonb/plotly
Package plotly uses GoNB plotly support (`github.com/janpfeifer/gonb/gonbui/plotly`) to plot both on dynamic plots while training or to quickly plot the results of a previously saved plot results in a checkpoints directory.
|
Package plotly uses GoNB plotly support (`github.com/janpfeifer/gonb/gonbui/plotly`) to plot both on dynamic plots while training or to quickly plot the results of a previously saved plot results in a checkpoints directory. |
notebook/gonb/plots
Package plots define common types and utilities to the different plot libraries.
|
Package plots define common types and utilities to the different plot libraries. |
ogbnmag
Package ogbnmag provides `Download` method for the corresponding dataset, and some dataset tools
|
Package ogbnmag provides `Download` method for the corresponding dataset, and some dataset tools |
ogbnmag/fnn
Package fnn implements a feed-forward neural network for the OGBN-MAG problem.
|
Package fnn implements a feed-forward neural network for the OGBN-MAG problem. |
ogbnmag/gnn
Package gnn implements a generic GNN modeling based on [TF-GNN MtAlbis].
|
Package gnn implements a generic GNN modeling based on [TF-GNN MtAlbis]. |
oxfordflowers102
Package oxfordflowers102 provides tools to download and cache the dataset and a `train.Dataset` implementation that can be used to train models using GoMLX (http://github.com/gomlx/gomlx/).
|
Package oxfordflowers102 provides tools to download and cache the dataset and a `train.Dataset` implementation that can be used to train models using GoMLX (http://github.com/gomlx/gomlx/). |
oxfordflowers102/diffusion
Package diffusion contains an example diffusion model, trained on Oxford Flowers 102 dataset.
|
Package diffusion contains an example diffusion model, trained on Oxford Flowers 102 dataset. |
experimental
|
|
nest
Package nest implements the Nest generic container, used for generic inputs and outputs in GoMLX.
|
Package nest implements the Nest generic container, used for generic inputs and outputs in GoMLX. |
Package graph is the core package for GoMLX.
|
Package graph is the core package for GoMLX. |
graphtest
Package graphtest holds test utilities for packages that depend on the graph package.
|
Package graphtest holds test utilities for packages that depend on the graph package. |
nanlogger
Package nanlogger collects `graph.Node` objects to monitor for `NaN` ("not-a-number") or `Inf` (infinity) values.
|
Package nanlogger collects `graph.Node` objects to monitor for `NaN` ("not-a-number") or `Inf` (infinity) values. |
ml
|
|
context
Package context defines the Context and Variable types: Context organizes variablesMap and variablesMap manages the storage of values typically used as variablesMap.
|
Package context defines the Context and Variable types: Context organizes variablesMap and variablesMap manages the storage of values typically used as variablesMap. |
context/checkpoints
Package checkpoints implements checkpoint management: saving and loading of checkpoints.
|
Package checkpoints implements checkpoint management: saving and loading of checkpoints. |
context/ctxtest
Package ctxtest holds test utilities for packages that depend on context package.
|
Package ctxtest holds test utilities for packages that depend on context package. |
context/initializers
Package initializers include several weight initializers, to be used with context.
|
Package initializers include several weight initializers, to be used with context. |
data
Package data is a collection of tools that facilitate data loading and preprocessing.
|
Package data is a collection of tools that facilitate data loading and preprocessing. |
data/hdf5
Package hdf5 provides a trivial API to access HDF5 file contents.
|
Package hdf5 provides a trivial API to access HDF5 file contents. |
layers
Package layers holds a collection of common modeling layers.
|
Package layers holds a collection of common modeling layers. |
train
Package train holds tools to help run a training loop.
|
Package train holds tools to help run a training loop. |
train/commandline
Package commandline contains convenience UI training tools for the command line.
|
Package commandline contains convenience UI training tools for the command line. |
train/losses
Package losses have several standard losses that implement train.LossFn interface.
|
Package losses have several standard losses that implement train.LossFn interface. |
train/metrics
Package metrics holds a library of metrics and defines
|
Package metrics holds a library of metrics and defines |
train/optimizers
Package optimizers implements a collection of ML optimizers, that can be used by train.Trainer, or by themselves.
|
Package optimizers implements a collection of ML optimizers, that can be used by train.Trainer, or by themselves. |
models
|
|
inceptionv3
Package inceptionv3 provides a pre-trained InceptionV3 model, or simply it's structure.
|
Package inceptionv3 provides a pre-trained InceptionV3 model, or simply it's structure. |
Package types is mostly a top level directory for GoMLX important types.
|
Package types is mostly a top level directory for GoMLX important types. |
exceptions
Package exceptions provides helper functions to leverage Go's `panic`, `recover` and `defer` as an "exceptions" system.
|
Package exceptions provides helper functions to leverage Go's `panic`, `recover` and `defer` as an "exceptions" system. |
keepalive
Package keepalive provides a simple Acquire and Release mechanism to make sure data is kept alive in between.
|
Package keepalive provides a simple Acquire and Release mechanism to make sure data is kept alive in between. |
shapes
Package shapes defines Shape and DType and associated tools.
|
Package shapes defines Shape and DType and associated tools. |
slices
Package slices are
|
Package slices are |
tensor
Package tensor provides a `Tensor` interface with 2 different implementations: `Local` and `Device`, they differ on where their values are stored: in the local (host) CPU, or on an accelerator device (TPU, GPU, but could be also the CPU if no accelerators are available).
|
Package tensor provides a `Tensor` interface with 2 different implementations: `Local` and `Device`, they differ on where their values are stored: in the local (host) CPU, or on an accelerator device (TPU, GPU, but could be also the CPU if no accelerators are available). |
tensor/image
Package image provides several functions to transform images back and forth from tensors.
|
Package image provides several functions to transform images back and forth from tensors. |
Package xla wraps the XLA functionality, plus extra dependencies.
|
Package xla wraps the XLA functionality, plus extra dependencies. |
Click to show internal directories.
Click to hide internal directories.