Skip to content

Commit 1a5bd75

Browse files
SiegeLordExtensorflower-gardener
authored andcommitted
Add a directory for Adaptive MALT project.
PiperOrigin-RevId: 473793633
1 parent cca6ecf commit 1a5bd75

File tree

4 files changed

+777
-0
lines changed

4 files changed

+777
-0
lines changed

discussion/adaptive_malt/BUILD

Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
1+
# Copyright 2022 The TensorFlow Probability Authors.
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
# ============================================================================
15+
# Adaptive MALT files.
16+
17+
# [internal] load pytype.bzl (pytype_strict_library)
18+
load("//third_party/py/pytest:pytest_defs.bzl", "pytest_test")
19+
20+
licenses(["notice"])
21+
22+
# pytype_strict
23+
py_library(
24+
name = "adaptive_malt",
25+
srcs = ["adaptive_malt.py"],
26+
deps = [
27+
# gin dep,
28+
# immutabledict dep,
29+
# jax dep,
30+
"//fun_mc:using_jax",
31+
"//inference_gym:using_jax",
32+
"//tensorflow_probability/substrates:jax",
33+
],
34+
)
35+
36+
pytest_test(
37+
name = "adaptive_malt_test",
38+
srcs = ["adaptive_malt_test.py"],
39+
deps = [
40+
":adaptive_malt",
41+
# absl/testing:absltest dep,
42+
# absl/testing:parameterized dep,
43+
# inference_gym:using_jax dep,
44+
# jax dep,
45+
# numpy dep,
46+
"//inference_gym:using_jax",
47+
"//tensorflow_probability/substrates:jax",
48+
],
49+
)

discussion/adaptive_malt/README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
# Adaptive MALT
2+
3+
This directory contains the implementation of the adaptive MALT algorithm.

0 commit comments

Comments
 (0)