Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ml-explore/mlx-data
/ functions
Functions
662 in github.com/ml-explore/mlx-data
⨍
Functions
662
◇
Types & classes
165
Method
build_extension
(self, ext)
setup.py:44
Method
can_merge
mlx/data/core/BPETokenizer.h:23
Method
cancel_prefetch
mlx/data/core/FileFetcher.cpp:143
Method
check_credentials
mlx/data/core/AWSFileFetcher.cpp:138
Method
check_samples_
mlx/data/buffer/FromVector.cpp:36
Method
collate_fn
(samples)
benchmarks/comparative/librispeech/pytorch.py:36
Method
concretize
mlx/data/Buffer.cpp:124
Method
contains
mlx/data/core/TARReader.cpp:179
Method
csv_reader_from_key
mlx/data/Stream.cpp:53
Method
decode_
mlx/data/core/video/VideoFFMPEG.cpp:441
Function
dl_iterate_callback
mlx/data/core/ThreadController.cpp:24
Function
document_iterator
Extract documents from wikitext raw file.
benchmarks/comparative/wikitext/tfds.py:12
Function
document_iterator
Extract documents from wikitext raw file.
benchmarks/comparative/wikitext/mlx_data.py:13
Method
dot_graph
mlx/data/core/Graph.cpp:32
Method
dynamic_batch
mlx/data/Stream.cpp:75
Method
dynamic_batch_
mlx/data/buffer/DynamicBatch.cpp:68
Function
file_to_label
(x)
python/mlx/data/datasets/imagenet.py:223
Function
files_and_classes
(root: Path)
benchmarks/comparative/librispeech/tfds.py:13
Function
files_from_tar
mlx/data/Buffer.cpp:139
Method
fill_queue_
mlx/data/core/FileFetcher.cpp:23
Method
filter_by_shape
mlx/data/Dataset.cpp:37
Method
filter_by_shape_if
mlx/data/Dataset.cpp:46
Method
filter_key_if
mlx/data/Dataset.cpp:65
Method
find_from_start_node
mlx/data/core/Graph.cpp:261
Method
find_from_start_node
mlx/data/core/Graph.cpp:304
Method
finish_background_tasks
mlx/data/stream/Buffered.cpp:99
Method
forward_
mlx/data/core/Tokenizer.cpp:207
Function
frames
Return the frame count of the image. Requires that `verifyVideo()` be called previously.
mlx/data/core/video/Video.h:52
Method
generate_random_crop_
mlx/data/op/ImageTransform.cpp:135
Method
get
mlx/data/buffer/FilesFromTAR.cpp:20
Method
get
mlx/data/buffer/Transform.cpp:21
Method
get
mlx/data/buffer/Buffer.cpp:11
Method
get
mlx/data/buffer/Append.cpp:14
Method
get
mlx/data/buffer/Batch.cpp:52
Method
get
mlx/data/buffer/Partition.cpp:27
Method
get_size
mlx/data/core/AWSFileFetcher.cpp:184
Method
get_tar_reader_
mlx/data/op/ReadFromTAR.cpp:32
Function
gpt2_byte_map
Return the byte map used for preprocessing the text for GPT-2 tokenizers. See https://github.com/openai/gpt-2/blob/master/src/encoder.py for deta
python/mlx/data/tokenizer_helpers.py:266
Function
height
Return the height of the image. Requires that `verifyVideo()` be called previously.
mlx/data/core/video/Video.h:40
Method
image_center_crop_if
mlx/data/Dataset.cpp:84
Method
image_channel_reduction
mlx/data/Dataset.cpp:98
Method
image_channel_reduction_if
mlx/data/Dataset.cpp:107
Method
image_random_area_crop
mlx/data/Dataset.cpp:121
Method
image_random_area_crop_if
mlx/data/Dataset.cpp:132
Method
image_random_crop
mlx/data/Dataset.cpp:148
Method
image_random_crop_if
mlx/data/Dataset.cpp:157
Method
image_random_h_flip
mlx/data/Dataset.cpp:171
Method
image_random_h_flip_if
mlx/data/Dataset.cpp:179
Method
image_resize
mlx/data/Dataset.cpp:192
Method
image_resize_if
mlx/data/Dataset.cpp:201
Method
image_resize_smallest_side_if
mlx/data/Dataset.cpp:224
Method
image_rotate
mlx/data/Dataset.cpp:238
Method
image_rotate_if
mlx/data/Dataset.cpp:247
Method
imemstream
mlx/data/core/imemstream.h:18
Function
info
mlx/data/core/image/ImageIO.cpp:26
Function
info
mlx/data/core/video/Video.cpp:59
Method
init
mlx/data/core/AWSFileFetcher.cpp:20
Method
init
finish initializing the `VideoReaderState`. Requires that the `formatContext_` be set.
mlx/data/core/video/VideoFFMPEG.cpp:263
Method
init_
mlx/data/Array.cpp:69
Method
init_
mlx/data/stream/LineReader.cpp:27
Method
is_final_node
mlx/data/core/Graph.cpp:157
Method
is_start_node
mlx/data/core/Graph.cpp:154
Method
is_valid_file
(self, filepath: str)
benchmarks/comparative/caltech101/pytorch.py:26
Function
iterate
(args, workers)
benchmarks/comparative/caltech101/pytorch.py:30
Function
iterate
(args, workers)
benchmarks/comparative/caltech101/tfds.py:47
Function
iterate
(args, workers)
benchmarks/comparative/caltech101/mlx_data.py:23
Function
iterate
(args, workers)
benchmarks/comparative/librispeech/pytorch.py:57
Function
iterate
(args, workers)
benchmarks/comparative/librispeech/tfds.py:63
Function
iterate
(args, workers)
benchmarks/comparative/librispeech/mlx_data.py:28
Function
iterate
(args, workers)
benchmarks/comparative/wikitext/pytorch.py:71
Function
iterate
(args)
benchmarks/comparative/wikitext/tfds.py:51
Function
iterate
(args, workers)
benchmarks/comparative/wikitext/mlx_data.py:41
Function
jpeg_error_custom_noop
mlx/data/core/Version.cpp:95
Function
jpeg_mem_fill_input_buffer
mlx/data/core/image/ImageJPEG.cpp:27
Function
jpeg_mem_init_source
mlx/data/core/image/ImageJPEG.cpp:25
Function
jpeg_mem_skip_input_data
mlx/data/core/image/ImageJPEG.cpp:35
Function
jpeg_mem_term_source
mlx/data/core/image/ImageJPEG.cpp:45
Function
levenshtein_arr
mlx/data/core/Levenshtein.cpp:66
Function
libs_version
mlx/data/core/Version.cpp:242
Function
load
mlx/data/core/image/ImageIO.cpp:10
Function
load
mlx/data/core/video/Video.cpp:10
Function
load_cifar10
Load a buffer with the CIFAR-10 dataset. Args: root (Path or str, optional): The directory to load/save the data. If none is
python/mlx/data/datasets/cifar.py:105
Function
load_cifar100
Load a buffer with the CIFAR-100 dataset. Args: root (Path or str, optional): The directory to load/save the data. If none is
python/mlx/data/datasets/cifar.py:128
Function
load_fashion_mnist
Load a buffer with the Fashion-MNIST dataset. If the data doesn't exist download it and save it for the next time. Args: root (Path
python/mlx/data/datasets/mnist.py:84
Method
load_file
mlx/data/Dataset.cpp:392
Method
load_file_if
mlx/data/Dataset.cpp:400
Method
load_image_if
mlx/data/Dataset.cpp:425
Function
load_imagenet
Load the ImageNet dataset from the downloaded archives. ImageNet cannot be automatically downloaded so you have to manually download it from
python/mlx/data/datasets/imagenet.py:124
Function
load_imagenet_metadata
Load the metadata for the imagenet classes.
python/mlx/data/datasets/imagenet.py:119
Function
load_images_from_folder
Load images from a folder. For a directory structure like the following .. code-block:: image_folder/ ...class_1/ ....
python/mlx/data/datasets/image_folder.py:8
Function
load_librispeech
Load the librispeech dataset directly from the TAR archive. Args: root (Path or str, optional): The The directory to load/save the data.
python/mlx/data/datasets/librispeech.py:108
Function
load_libritts_r
Load the LibriTTS-R dataset directly from the TAR archive. Args: root (Path or str, optional): The The directory to load/save the data. I
python/mlx/data/datasets/libritts_r.py:98
Function
load_mnist
Load a buffer with the MNIST dataset. If the data doesn't exist download it and save it for the next time. Args: root (Path or str,
python/mlx/data/datasets/mnist.py:70
Method
load_numpy
mlx/data/Dataset.cpp:442
Method
load_numpy_if
mlx/data/Dataset.cpp:452
Function
load_speechcommands
Load the Speech Commands (v0.0.2) [1] dataset directly from the TAR archive. Args: root (Path or str, optional): The The directory to loa
python/mlx/data/datasets/speechcommands.py:88
Method
load_video
mlx/data/Dataset.cpp:467
Method
load_video_if
mlx/data/Dataset.cpp:478
Function
load_wikitext_lines
Fetch the WikiText dataset and load it as a stream of lines. Args: root (Path or str, optional): The The directory to load/save the data.
python/mlx/data/datasets/wikitext.py:35
Method
log_run
(self, run_name, fn, *args, **kwargs)
benchmarks/utils.py:14
← previous
next →
401–500 of 662, ranked by callers